Compare commits
23 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5b4c1c1471 | ||
|
|
fc3edd7b8f | ||
|
|
93aa0ee1ac | ||
|
|
42378714a0 | ||
|
|
759f022c98 | ||
|
|
5af4d481f2 | ||
|
|
221bb5beb7 | ||
|
|
5a89e86120 | ||
|
|
e351a92680 | ||
|
|
83cd0cfea3 | ||
|
|
9e29f611df | ||
|
|
d8ea4b70a9 | ||
|
|
150ddac2c0 | ||
|
|
5108ad1f6b | ||
|
|
14766ae171 | ||
|
|
ccd91a7b6f | ||
|
|
a894ad85ba | ||
|
|
1da05f5cae | ||
|
|
562f528c01 | ||
|
|
840b0e6205 | ||
| cb4d01c612 | |||
| e6344f1a92 | |||
|
|
3082c261c7 |
@@ -1,6 +1,6 @@
|
|||||||
# This file is machine-generated - editing it directly is not advised
|
# This file is machine-generated - editing it directly is not advised
|
||||||
|
|
||||||
julia_version = "1.11.2"
|
julia_version = "1.11.4"
|
||||||
manifest_format = "2.0"
|
manifest_format = "2.0"
|
||||||
project_hash = "75c6a269a13b222c106479d2177b05facfa23f74"
|
project_hash = "75c6a269a13b222c106479d2177b05facfa23f74"
|
||||||
|
|
||||||
@@ -310,7 +310,7 @@ version = "0.3.27+1"
|
|||||||
[[deps.OpenLibm_jll]]
|
[[deps.OpenLibm_jll]]
|
||||||
deps = ["Artifacts", "Libdl"]
|
deps = ["Artifacts", "Libdl"]
|
||||||
uuid = "05823500-19ac-5b8b-9628-191a04bc5112"
|
uuid = "05823500-19ac-5b8b-9628-191a04bc5112"
|
||||||
version = "0.8.1+2"
|
version = "0.8.1+4"
|
||||||
|
|
||||||
[[deps.OpenSpecFun_jll]]
|
[[deps.OpenSpecFun_jll]]
|
||||||
deps = ["Artifacts", "CompilerSupportLibraries_jll", "JLLWrappers", "Libdl", "Pkg"]
|
deps = ["Artifacts", "CompilerSupportLibraries_jll", "JLLWrappers", "Libdl", "Pkg"]
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
name = "GeneralUtils"
|
name = "GeneralUtils"
|
||||||
uuid = "c6c72f09-b708-4ac8-ac7c-2084d70108fe"
|
uuid = "c6c72f09-b708-4ac8-ac7c-2084d70108fe"
|
||||||
authors = ["tonaerospace <tonaerospace.etc@gmail.com>"]
|
authors = ["tonaerospace <tonaerospace.etc@gmail.com>"]
|
||||||
version = "0.2.2"
|
version = "0.3.0"
|
||||||
|
|
||||||
[deps]
|
[deps]
|
||||||
CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"
|
CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"
|
||||||
|
|||||||
@@ -74,6 +74,7 @@ mutable struct mqttClientInstance_v2 <: mqttClientInstance
|
|||||||
onMsgCallback::Function
|
onMsgCallback::Function
|
||||||
qos::MQTTClient.QOS
|
qos::MQTTClient.QOS
|
||||||
client::MQTTClient.Client
|
client::MQTTClient.Client
|
||||||
|
clientId::String
|
||||||
connection::MQTTClient.Connection
|
connection::MQTTClient.Connection
|
||||||
keepalivetopic::String
|
keepalivetopic::String
|
||||||
keepaliveChannel::Channel
|
keepaliveChannel::Channel
|
||||||
@@ -153,8 +154,9 @@ function mqttClientInstance_v2(
|
|||||||
keepaliveChannel::Channel, # used for checkMqttConnection(). user needs to specify because it has to be accessible by user-defined onMsgCallback()
|
keepaliveChannel::Channel, # used for checkMqttConnection(). user needs to specify because it has to be accessible by user-defined onMsgCallback()
|
||||||
onMsgCallback::Function
|
onMsgCallback::Function
|
||||||
;
|
;
|
||||||
|
clientId::String="NA",
|
||||||
mqttBrokerPort::Integer=1883,
|
mqttBrokerPort::Integer=1883,
|
||||||
keepalivetopic::String= "/keepalive/$(uuid4snakecase())",
|
keepaliveTopic::String= "/keepalive/$(uuid4snakecase())",
|
||||||
keepaliveCheckInterval::Integer=30,
|
keepaliveCheckInterval::Integer=30,
|
||||||
qos::MQTTClient.QOS=QOS_1,
|
qos::MQTTClient.QOS=QOS_1,
|
||||||
multiMsg::String="single", #[PENDING] bad design. this info should be stored in each msgMeta
|
multiMsg::String="single", #[PENDING] bad design. this info should be stored in each msgMeta
|
||||||
@@ -166,7 +168,9 @@ function mqttClientInstance_v2(
|
|||||||
for i in subtopic
|
for i in subtopic
|
||||||
MQTTClient.subscribe(client, i, onMsgCallback, qos=qos)
|
MQTTClient.subscribe(client, i, onMsgCallback, qos=qos)
|
||||||
end
|
end
|
||||||
MQTTClient.subscribe(client, keepalivetopic, onMsgCallback, qos=qos)
|
MQTTClient.subscribe(client, keepaliveTopic, onMsgCallback, qos=qos)
|
||||||
|
|
||||||
|
keepaliveTopic = clientId == "NA" ? keepaliveTopic : "/keepalive/$clientId"
|
||||||
|
|
||||||
instance = mqttClientInstance_v2(
|
instance = mqttClientInstance_v2(
|
||||||
mqttBrokerAddress,
|
mqttBrokerAddress,
|
||||||
@@ -176,8 +180,9 @@ function mqttClientInstance_v2(
|
|||||||
onMsgCallback,
|
onMsgCallback,
|
||||||
qos,
|
qos,
|
||||||
client,
|
client,
|
||||||
|
clientId,
|
||||||
connection,
|
connection,
|
||||||
keepalivetopic,
|
keepaliveTopic,
|
||||||
keepaliveChannel,
|
keepaliveChannel,
|
||||||
keepaliveCheckInterval,
|
keepaliveCheckInterval,
|
||||||
nothing,
|
nothing,
|
||||||
@@ -346,8 +351,6 @@ end
|
|||||||
-----
|
-----
|
||||||
"""
|
"""
|
||||||
function isMqttConnectionAlive(mqttInstance::T)::Bool where {T<:mqttClientInstance}
|
function isMqttConnectionAlive(mqttInstance::T)::Bool where {T<:mqttClientInstance}
|
||||||
|
|
||||||
starttime = Dates.now()
|
|
||||||
isconnectionalive = false
|
isconnectionalive = false
|
||||||
|
|
||||||
# ditch old keepalive msg is any
|
# ditch old keepalive msg is any
|
||||||
@@ -369,9 +372,9 @@ function isMqttConnectionAlive(mqttInstance::T)::Bool where {T<:mqttClientInstan
|
|||||||
|
|
||||||
publish(mqttInstance.client, keepaliveMsg[:msgMeta][:sendTopic],
|
publish(mqttInstance.client, keepaliveMsg[:msgMeta][:sendTopic],
|
||||||
JSON3.write(keepaliveMsg))
|
JSON3.write(keepaliveMsg))
|
||||||
|
|
||||||
timediff = 0
|
timediff = 0
|
||||||
while timediff < 5
|
starttime = Dates.now()
|
||||||
|
while timediff <= 5
|
||||||
timediff = timedifference(starttime, Dates.now(), "seconds")
|
timediff = timedifference(starttime, Dates.now(), "seconds")
|
||||||
if isready(mqttInstance.keepaliveChannel)
|
if isready(mqttInstance.keepaliveChannel)
|
||||||
incomingMsg = take!(mqttInstance.keepaliveChannel)
|
incomingMsg = take!(mqttInstance.keepaliveChannel)
|
||||||
@@ -393,7 +396,7 @@ end
|
|||||||
-----
|
-----
|
||||||
mqttInstanceDict::Dict{Symbol, Any}
|
mqttInstanceDict::Dict{Symbol, Any}
|
||||||
a dictionary contain mqtt instance. 1 per mqtt client.
|
a dictionary contain mqtt instance. 1 per mqtt client.
|
||||||
interval::Integer
|
keepaliveCheckInterval::Integer
|
||||||
time interval to check mqtt server in seconds
|
time interval to check mqtt server in seconds
|
||||||
|
|
||||||
Return\n
|
Return\n
|
||||||
@@ -413,7 +416,7 @@ end
|
|||||||
-----
|
-----
|
||||||
"""
|
"""
|
||||||
function checkMqttConnection!(mqttInstance::T;
|
function checkMqttConnection!(mqttInstance::T;
|
||||||
keepaliveCheckInterval::Union{Integer, Nothing}=nothing) where {T<:mqttClientInstance}
|
keepaliveCheckInterval::Union{Integer, Nothing}=nothing)::Union{Bool, Nothing} where {T<:mqttClientInstance}
|
||||||
|
|
||||||
interval = keepaliveCheckInterval !== nothing ? keepaliveCheckInterval : mqttInstance.keepaliveCheckInterval
|
interval = keepaliveCheckInterval !== nothing ? keepaliveCheckInterval : mqttInstance.keepaliveCheckInterval
|
||||||
|
|
||||||
@@ -425,29 +428,32 @@ function checkMqttConnection!(mqttInstance::T;
|
|||||||
Inf
|
Inf
|
||||||
end
|
end
|
||||||
|
|
||||||
isreconnect = false # this value is true if connection is disconnected
|
|
||||||
if intervaldiff > interval
|
if intervaldiff > interval
|
||||||
|
connectionStatusStart = isMqttConnectionAlive(mqttInstance) # a flag to note whether the connection status has changed from false to true
|
||||||
while true
|
while true
|
||||||
mqttConnStatus = isMqttConnectionAlive(mqttInstance)
|
mqttConnStatus = isMqttConnectionAlive(mqttInstance)
|
||||||
if mqttConnStatus == false
|
if mqttConnStatus == false
|
||||||
isreconnect = true
|
sleep(5) # wait
|
||||||
println("mqtt connection disconnected, attemping to reconnect $(Dates.now())")
|
println("MQTT connection disconnected, attemping to reconnect $(Dates.now()) at $(mqttInstance.mqttBrokerAddress):$(mqttInstance.mqttBrokerPort)")
|
||||||
# use new client to reconnect instead of the previous one because I don't want to modify MQTTClient.jl yet
|
# use new client to reconnect instead of the previous one because I don't want to modify MQTTClient.jl yet
|
||||||
mqttInstance.client, mqttInstance.connection =
|
mqttInstance.client, mqttInstance.connection =
|
||||||
MakeConnection(mqttInstance.mqttBrokerAddress,
|
MakeConnection(mqttInstance.mqttBrokerAddress,
|
||||||
mqttInstance.mqttBrokerPort)
|
mqttInstance.mqttBrokerPort)
|
||||||
connect(mqttInstance.client, mqttInstance.connection)
|
try
|
||||||
for topic in mqttInstance.subtopic
|
connect(mqttInstance.client, mqttInstance.connection)
|
||||||
subscribe(mqttInstance.client, topic, mqttInstance.onMsgCallback, qos=mqttInstance.qos)
|
for topic in mqttInstance.subtopic
|
||||||
|
subscribe(mqttInstance.client, topic, mqttInstance.onMsgCallback, qos=mqttInstance.qos)
|
||||||
|
end
|
||||||
|
MQTTClient.subscribe(mqttInstance.client, mqttInstance.keepalivetopic, mqttInstance.onMsgCallback, qos=mqttInstance.qos)
|
||||||
|
catch
|
||||||
|
println("Failed to reconnect MQTT broker at $(mqttInstance.mqttBrokerAddress):$(mqttInstance.mqttBrokerPort) $(Dates.now())")
|
||||||
end
|
end
|
||||||
MQTTClient.subscribe(mqttInstance.client, mqttInstance.keepalivetopic, mqttInstance.onMsgCallback, qos=mqttInstance.qos)
|
|
||||||
sleep(1) # wait before checking connection again
|
|
||||||
else
|
else
|
||||||
mqttInstance.lastTimeMqttConnCheck = Dates.now()
|
mqttInstance.lastTimeMqttConnCheck = Dates.now()
|
||||||
if isreconnect
|
if connectionStatusStart != mqttConnStatus
|
||||||
println("connected to mqtt broker")
|
println("Reconnected to MQTT broker at $(mqttInstance.mqttBrokerAddress):$(mqttInstance.mqttBrokerPort) $(Dates.now())")
|
||||||
end
|
end
|
||||||
return isreconnect
|
return mqttConnStatus
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
else
|
else
|
||||||
@@ -567,8 +573,9 @@ julia> success, error, response = GeneralUtils.sendReceiveMqttMsg(outgoingMsg)
|
|||||||
# Signature
|
# Signature
|
||||||
"""
|
"""
|
||||||
function sendReceiveMqttMsg(outgoingMsg::Dict{Symbol, T};
|
function sendReceiveMqttMsg(outgoingMsg::Dict{Symbol, T};
|
||||||
timeout::Integer=60, maxattempt::Integer=1)::NamedTuple where {T<:Any}
|
connectiontimeout::Integer=600, responsetimeout::Integer=60, responsemaxattempt::Integer=1)::NamedTuple where {T<:Any}
|
||||||
mqttMsgReceiveTopic = "/GeneralUtils_sendReceiveMqttMsg/$(outgoingMsg[:msgMeta][:senderId])"
|
senderId = outgoingMsg[:msgMeta][:senderId]
|
||||||
|
mqttMsgReceiveTopic = "/GeneralUtils_sendReceiveMqttMsg/$senderId"
|
||||||
mqttMsgReceiveChannel = (ch1=Channel(8),)
|
mqttMsgReceiveChannel = (ch1=Channel(8),)
|
||||||
keepaliveChannel = Channel(8)
|
keepaliveChannel = Channel(8)
|
||||||
outgoingMsg[:msgMeta][:replyTopic] = mqttMsgReceiveTopic
|
outgoingMsg[:msgMeta][:replyTopic] = mqttMsgReceiveTopic
|
||||||
@@ -576,45 +583,72 @@ function sendReceiveMqttMsg(outgoingMsg::Dict{Symbol, T};
|
|||||||
# Define the callback for receiving messages.
|
# Define the callback for receiving messages.
|
||||||
function onMsgCallback(topic, payload)
|
function onMsgCallback(topic, payload)
|
||||||
jobj = JSON3.read(String(payload))
|
jobj = JSON3.read(String(payload))
|
||||||
onMsg = copy(jobj)
|
incomingMqttMsg = copy(jobj)
|
||||||
put!(mqttMsgReceiveChannel[:ch1], onMsg)
|
|
||||||
|
if occursin("GeneralUtils_sendReceiveMqttMsg", topic)
|
||||||
|
put!(mqttMsgReceiveChannel[:ch1], incomingMqttMsg)
|
||||||
|
elseif occursin("keepalive", topic)
|
||||||
|
put!(keepaliveChannel, incomingMqttMsg)
|
||||||
|
else
|
||||||
|
println("undefined condition ", @__FILE__, " ", @__LINE__)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
mqttInstance = mqttClientInstance_v2(
|
mqttInstance = nothing
|
||||||
outgoingMsg[:msgMeta][:mqttBrokerAddress],
|
attempt = 0
|
||||||
[mqttMsgReceiveTopic],
|
starttime = Dates.now()
|
||||||
mqttMsgReceiveChannel,
|
endtime = starttime + Second(connectiontimeout)
|
||||||
keepaliveChannel,
|
errormsg = nothing
|
||||||
onMsgCallback;
|
while true
|
||||||
mqttBrokerPort=outgoingMsg[:msgMeta][:mqttBrokerPort]
|
timenow = Dates.now()
|
||||||
)
|
timepass = timedifference(starttime, timenow, "seconds")
|
||||||
|
timeleft = timedifference(timenow, endtime, "seconds")
|
||||||
|
|
||||||
response = sendReceiveMqttMsg(mqttInstance, :ch1, outgoingMsg; timeout=timeout, maxattempt=maxattempt)
|
if timepass <= connectiontimeout
|
||||||
|
attempt += 1
|
||||||
|
attempt > 1 ? println("Attempt $attempt to connect to MQTT broker. Timed out in $timeleft seconds. $errormsg") : nothing
|
||||||
|
|
||||||
|
try
|
||||||
|
mqttInstance = mqttClientInstance_v2(
|
||||||
|
outgoingMsg[:msgMeta][:mqttBrokerAddress],
|
||||||
|
[mqttMsgReceiveTopic],
|
||||||
|
mqttMsgReceiveChannel,
|
||||||
|
keepaliveChannel,
|
||||||
|
onMsgCallback;
|
||||||
|
mqttBrokerPort=outgoingMsg[:msgMeta][:mqttBrokerPort],
|
||||||
|
clientId=senderId
|
||||||
|
)
|
||||||
|
break
|
||||||
|
catch e
|
||||||
|
errormsg = e
|
||||||
|
sleep(5)
|
||||||
|
end
|
||||||
|
else
|
||||||
|
println("Failed to instantiate MQTT client after $timepass seconds. $errormsg")
|
||||||
|
return nothing
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
response = sendReceiveMqttMsg(mqttInstance, :ch1, outgoingMsg;
|
||||||
|
responsetimeout=responsetimeout, responsemaxattempt=responsemaxattempt)
|
||||||
try disconnect(mqttInstance.client) catch end
|
try disconnect(mqttInstance.client) catch end
|
||||||
|
|
||||||
return response
|
return response
|
||||||
end
|
end
|
||||||
function sendReceiveMqttMsg(mqttInstance::mqttClientInstance_v2, receivechannel::Symbol,
|
function sendReceiveMqttMsg(mqttInstance::mqttClientInstance_v2, receivechannel::Symbol,
|
||||||
outgoingMsg::Dict{Symbol, T}; timeout::Integer=60, maxattempt::Integer=1
|
outgoingMsg::Dict{Symbol, T}; responsetimeout::Integer=60, responsemaxattempt::Integer=1
|
||||||
)::NamedTuple where {T<:Any}
|
)::NamedTuple where {T<:Any}
|
||||||
|
|
||||||
timepass = nothing
|
timepass = nothing
|
||||||
attempts = 0
|
attempts = 1
|
||||||
while attempts <= maxattempt
|
while attempts <= responsemaxattempt
|
||||||
attempts += 1
|
|
||||||
if attempts > 1
|
|
||||||
println("\n<sendReceiveMqttMsg()> attempts $attempts/$maxattempt ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
|
|
||||||
pprintln(outgoingMsg)
|
|
||||||
println("</sendReceiveMqttMsg()> attempts $attempts/$maxattempt ", @__FILE__, ":", @__LINE__, " $(Dates.now())\n")
|
|
||||||
end
|
|
||||||
|
|
||||||
sendMqttMsg(mqttInstance, outgoingMsg)
|
sendMqttMsg(mqttInstance, outgoingMsg)
|
||||||
|
|
||||||
starttime = Dates.now()
|
starttime = Dates.now()
|
||||||
while true
|
while true
|
||||||
timepass = timedifference(starttime, Dates.now(), "seconds")
|
timepass = timedifference(starttime, Dates.now(), "seconds")
|
||||||
|
|
||||||
if timepass <= timeout
|
if timepass <= responsetimeout
|
||||||
if isready(mqttInstance.msgReceiveChannel[receivechannel])
|
if isready(mqttInstance.msgReceiveChannel[receivechannel])
|
||||||
incomingMsg = take!(mqttInstance.msgReceiveChannel[receivechannel])
|
incomingMsg = take!(mqttInstance.msgReceiveChannel[receivechannel])
|
||||||
|
|
||||||
@@ -632,6 +666,13 @@ function sendReceiveMqttMsg(mqttInstance::mqttClientInstance_v2, receivechannel:
|
|||||||
end
|
end
|
||||||
sleep(1)
|
sleep(1)
|
||||||
end
|
end
|
||||||
|
if attempts > 1
|
||||||
|
println("\n<sendReceiveMqttMsg()> attempts $attempts/$responsemaxattempt ", @__FILE__, ":", @__LINE__, " $(Dates.now())")
|
||||||
|
pprintln(outgoingMsg)
|
||||||
|
println("</sendReceiveMqttMsg()> attempts $attempts/$responsemaxattempt ", @__FILE__, ":", @__LINE__, " $(Dates.now())\n")
|
||||||
|
checkMqttConnection!(mqttInstance, keepaliveCheckInterval=5)
|
||||||
|
end
|
||||||
|
attempts += 1
|
||||||
end
|
end
|
||||||
|
|
||||||
return (success=false,
|
return (success=false,
|
||||||
|
|||||||
267
src/llmUtil.jl
267
src/llmUtil.jl
@@ -1,6 +1,6 @@
|
|||||||
module llmUtil
|
module llmUtil
|
||||||
|
|
||||||
export formatLLMtext, formatLLMtext_llama3instruct, jsoncorrection
|
export formatLLMtext, formatLLMtext_llama3instruct, jsoncorrection, deFormatLLMtext, extractthink
|
||||||
|
|
||||||
using UUIDs, JSON3, Dates
|
using UUIDs, JSON3, Dates
|
||||||
using GeneralUtils
|
using GeneralUtils
|
||||||
@@ -43,7 +43,7 @@ julia> formattedtext = YiemAgent.formatLLMtext_llama3instruct(d[:name], d[:text]
|
|||||||
Signature
|
Signature
|
||||||
"""
|
"""
|
||||||
function formatLLMtext_llama3instruct(name::T, text::T;
|
function formatLLMtext_llama3instruct(name::T, text::T;
|
||||||
assistantStarter::Bool=true) where {T<:AbstractString}
|
assistantStarter::Bool=false) where {T<:AbstractString}
|
||||||
formattedtext =
|
formattedtext =
|
||||||
if name == "system"
|
if name == "system"
|
||||||
"""
|
"""
|
||||||
@@ -68,28 +68,10 @@ function formatLLMtext_llama3instruct(name::T, text::T;
|
|||||||
|
|
||||||
return formattedtext
|
return formattedtext
|
||||||
end
|
end
|
||||||
# function formatLLMtext_llama3instruct(name::T, text::T) where {T<:AbstractString}
|
|
||||||
# formattedtext =
|
|
||||||
# if name == "system"
|
|
||||||
# """
|
|
||||||
# <|begin_of_text|>
|
|
||||||
# <|start_header_id|>$name<|end_header_id|>
|
|
||||||
# $text
|
|
||||||
# <|eot_id|>
|
|
||||||
# """
|
|
||||||
# else
|
|
||||||
# """
|
|
||||||
# <|start_header_id|>$name<|end_header_id|>
|
|
||||||
# $text
|
|
||||||
# <|eot_id|>
|
|
||||||
# """
|
|
||||||
# end
|
|
||||||
|
|
||||||
# return formattedtext
|
|
||||||
# end
|
|
||||||
|
|
||||||
function formatLLMtext_qwen(name::T, text::T;
|
function formatLLMtext_qwen(name::T, text::T;
|
||||||
assistantStarter::Bool=true) where {T<:AbstractString}
|
assistantStarter::Bool=false) where {T<:AbstractString}
|
||||||
formattedtext =
|
formattedtext =
|
||||||
if name == "system"
|
if name == "system"
|
||||||
"""
|
"""
|
||||||
@@ -116,14 +98,94 @@ function formatLLMtext_qwen(name::T, text::T;
|
|||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
""" Convert a chat messages in vector of dictionary into LLM model instruct format.
|
function formatLLMtext_qwen3(name::T, text::T;
|
||||||
|
assistantStarter::Bool=false) where {T<:AbstractString}
|
||||||
|
formattedtext =
|
||||||
|
if name == "system"
|
||||||
|
"""
|
||||||
|
<|im_start|>$name
|
||||||
|
$text
|
||||||
|
<|im_end|>
|
||||||
|
"""
|
||||||
|
else
|
||||||
|
"""
|
||||||
|
<|im_start|>$name
|
||||||
|
$text
|
||||||
|
<|im_end|>
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
if assistantStarter
|
||||||
|
formattedtext *=
|
||||||
|
"""
|
||||||
|
<|im_start|>assistant
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
return formattedtext
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
function formatLLMtext_phi4(name::T, text::T;
|
||||||
|
assistantStarter::Bool=false) where {T<:AbstractString}
|
||||||
|
formattedtext =
|
||||||
|
if name == "system"
|
||||||
|
"""
|
||||||
|
<|system|>
|
||||||
|
$text
|
||||||
|
<|end|>
|
||||||
|
"""
|
||||||
|
else
|
||||||
|
"""
|
||||||
|
<|assistant|>
|
||||||
|
$text
|
||||||
|
<|end|>
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
if assistantStarter
|
||||||
|
formattedtext *=
|
||||||
|
"""
|
||||||
|
<|assistant|>
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
return formattedtext
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
function formatLLMtext_granite3(name::T, text::T;
|
||||||
|
assistantStarter::Bool=false) where {T<:AbstractString}
|
||||||
|
formattedtext =
|
||||||
|
if name == "system"
|
||||||
|
"""
|
||||||
|
<|start_of_role|>system<|end_of_role|>{$text}<|end_of_text|>
|
||||||
|
"""
|
||||||
|
else
|
||||||
|
"""
|
||||||
|
<|start_of_role|>$name<|end_of_role|>{$text}<|end_of_text|>
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
if assistantStarter
|
||||||
|
formattedtext *=
|
||||||
|
"""
|
||||||
|
<|start_of_role|>assistant<|end_of_role|>{
|
||||||
|
"""
|
||||||
|
end
|
||||||
|
|
||||||
|
return formattedtext
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
""" Convert a vector of chat message dictionaries into LLM model instruct format.
|
||||||
|
|
||||||
# Arguments
|
# Arguments
|
||||||
- `messages::Vector{Dict{Symbol, T}}`
|
- `messages::Vector{Dict{Symbol, T}}`
|
||||||
message owner name e.f. "system", "user" or "assistant"
|
A vector of dictionaries where each dictionary contains the keys `:name` (the name of the message owner) and `:text` (the text of the message).
|
||||||
- `formatname::T`
|
- `formatname::T`
|
||||||
format name to be used
|
The name of the format to be used for converting the chat messages.
|
||||||
|
|
||||||
# Return
|
# Return
|
||||||
- `formattedtext::String`
|
- `formattedtext::String`
|
||||||
text formatted to model format
|
text formatted to model format
|
||||||
@@ -140,31 +202,137 @@ julia> chatmessage = [
|
|||||||
julia> formattedtext = YiemAgent.formatLLMtext(chatmessage, "llama3instruct")
|
julia> formattedtext = YiemAgent.formatLLMtext(chatmessage, "llama3instruct")
|
||||||
"<|begin_of_text|>\n <|start_header_id|>system<|end_header_id|>\n You are a helpful, respectful and honest assistant.\n <|eot_id|>\n <|start_header_id|>user<|end_header_id|>\n list me all planets in our solar system.\n <|eot_id|>\n <|start_header_id|>assistant<|end_header_id|>\n I'm sorry. I don't know. You tell me.\n <|eot_id|>\n"
|
"<|begin_of_text|>\n <|start_header_id|>system<|end_header_id|>\n You are a helpful, respectful and honest assistant.\n <|eot_id|>\n <|start_header_id|>user<|end_header_id|>\n list me all planets in our solar system.\n <|eot_id|>\n <|start_header_id|>assistant<|end_header_id|>\n I'm sorry. I don't know. You tell me.\n <|eot_id|>\n"
|
||||||
```
|
```
|
||||||
|
|
||||||
# Signature
|
|
||||||
"""
|
"""
|
||||||
function formatLLMtext(messages::Vector{Dict{Symbol, T}}; formatname::String="llama3instruct"
|
function formatLLMtext(messages::Vector{Dict{Symbol, T}}, formatname::String
|
||||||
)::String where {T<:Any}
|
)::String where {T<:AbstractString}
|
||||||
f = if formatname == "llama3instruct"
|
f =
|
||||||
formatLLMtext_llama3instruct
|
if formatname == "llama3instruct"
|
||||||
elseif formatname == "mistral"
|
formatLLMtext_llama3instruct
|
||||||
# not define yet
|
elseif formatname == "mistral"
|
||||||
elseif formatname == "phi3instruct"
|
# not define yet
|
||||||
# not define yet
|
elseif formatname == "phi3instruct"
|
||||||
elseif formatname == "qwen"
|
# not define yet
|
||||||
formatLLMtext_qwen
|
elseif formatname == "qwen"
|
||||||
else
|
formatLLMtext_qwen
|
||||||
error("$formatname template not define yet")
|
elseif formatname == "qwen3"
|
||||||
end
|
formatLLMtext_qwen3
|
||||||
|
elseif formatname == "phi4"
|
||||||
|
formatLLMtext_phi4
|
||||||
|
elseif formatname == "granite3"
|
||||||
|
formatLLMtext_granite3
|
||||||
|
else
|
||||||
|
error("$formatname template not define yet")
|
||||||
|
end
|
||||||
|
|
||||||
str = ""
|
str = ""
|
||||||
for t in messages
|
for (i, t) in enumerate(messages)
|
||||||
str *= f(t[:name], t[:text])
|
if i < length(messages)
|
||||||
|
str *= f(t[:name], t[:text])
|
||||||
|
else
|
||||||
|
str *= f(t[:name], t[:text]; assistantStarter=true)
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
return str
|
return str
|
||||||
end
|
end
|
||||||
|
|
||||||
|
""" Revert LLM-format response back into regular text.
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `text::String`
|
||||||
|
The LLM formatted string to be converted.
|
||||||
|
|
||||||
|
# Return
|
||||||
|
- `normalText::String`
|
||||||
|
The original plain text extracted from the given LLM-formatted string.
|
||||||
|
|
||||||
|
# Example
|
||||||
|
```jldoctest
|
||||||
|
julia> using Revise
|
||||||
|
julia> using YiemAgent
|
||||||
|
julia> response = "<|begin_of_text|>This is a sample system instruction.<|eot_id|>"
|
||||||
|
julia> normalText = YiemAgent.deFormatLLMtext(response, "granite3")
|
||||||
|
"This is a sample system instruction."
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
function deFormatLLMtext(text::String, formatname::String; includethink::Bool=false
|
||||||
|
)::String
|
||||||
|
f =
|
||||||
|
if formatname == "granite3"
|
||||||
|
deFormatLLMtext_granite3
|
||||||
|
elseif formatname == "qwen3"
|
||||||
|
deFormatLLMtext_qwen3
|
||||||
|
else
|
||||||
|
error("$formatname template not define yet")
|
||||||
|
end
|
||||||
|
|
||||||
|
r = f(text)
|
||||||
|
result = r === nothing ? text : r
|
||||||
|
return result
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
""" Revert LLM-format response back into regular text for Granite 3 format.
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `text::String`
|
||||||
|
The LLM formatted string to be converted.
|
||||||
|
|
||||||
|
# Return
|
||||||
|
- `normalText::Union{Nothing, String}`
|
||||||
|
The original plain text extracted from the given LLM-formatted string.
|
||||||
|
Returns nothing if the text is not in Granite 3 format.
|
||||||
|
|
||||||
|
# Example
|
||||||
|
```jldoctest
|
||||||
|
julia> using Revise
|
||||||
|
julia> using YiemAgent
|
||||||
|
julia> response = "{This is a sample LLM response.}"
|
||||||
|
julia> normalText = YiemAgent.deFormatLLMtext(response, "granite3")
|
||||||
|
"This is a sample LLM response."
|
||||||
|
"""
|
||||||
|
function deFormatLLMtext_granite3(text::String)::Union{Nothing, String}
|
||||||
|
# check if '{' and '}' are in the text because it's a special format for the LLM response
|
||||||
|
if contains(text, "<|im_start|>assistant")
|
||||||
|
# get the text between '{' and '}'
|
||||||
|
text_between_braces = GeneralUtils.extractTextBetweenCharacter(text, '{', '}')[1]
|
||||||
|
return text_between_braces
|
||||||
|
elseif text[end] == '}'
|
||||||
|
text = "{$text"
|
||||||
|
text_between_braces = GeneralUtils.extractTextBetweenCharacter(text, '{', '}')[1]
|
||||||
|
else
|
||||||
|
return nothing
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
function deFormatLLMtext_qwen3(text::String)::Union{Nothing, String}
|
||||||
|
return text
|
||||||
|
end
|
||||||
|
|
||||||
|
# function deFormatLLMtext_qwen3(text::String; includethink::Bool=false)::Union{Nothing, String}
|
||||||
|
# think = nothing
|
||||||
|
# str = nothing
|
||||||
|
|
||||||
|
# if occursin("<think>", text)
|
||||||
|
# r = GeneralUtils.extractTextBetweenString(text, "<think>", "</think>")
|
||||||
|
# if r[:success]
|
||||||
|
# think = r[:text]
|
||||||
|
# end
|
||||||
|
# str = string(split(text, "</think>")[2])
|
||||||
|
# end
|
||||||
|
|
||||||
|
# if includethink == true && occursin("<think>", text)
|
||||||
|
# result = "ModelThought: $think $str"
|
||||||
|
# return result
|
||||||
|
# elseif includethink == false && occursin("<think>", text)
|
||||||
|
# result = str
|
||||||
|
# return result
|
||||||
|
# else
|
||||||
|
# return text
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
|
||||||
|
|
||||||
""" Attemp to correct LLM response's incorrect JSON response.
|
""" Attemp to correct LLM response's incorrect JSON response.
|
||||||
|
|
||||||
@@ -255,7 +423,20 @@ function jsoncorrection(config::T1, input::T2, correctJsonExample::T3;
|
|||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
|
function extractthink(text::String)
|
||||||
|
think = nothing
|
||||||
|
str = nothing
|
||||||
|
if occursin("<think>", text)
|
||||||
|
r = GeneralUtils.extractTextBetweenString(text, "<think>", "</think>")
|
||||||
|
if r[:success]
|
||||||
|
think = r[:text]
|
||||||
|
end
|
||||||
|
str = string(split(text, "</think>")[2])
|
||||||
|
else
|
||||||
|
str = text
|
||||||
|
end
|
||||||
|
return think, str
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
287
src/util.jl
287
src/util.jl
@@ -6,7 +6,8 @@ export timedifference, showstracktrace, findHighestIndexKey, uuid4snakecase, rep
|
|||||||
dfToString, dataframe_to_json_list, dictToString, dictToString_noKey,
|
dfToString, dataframe_to_json_list, dictToString, dictToString_noKey,
|
||||||
dictToString_numbering, extract_triple_backtick_text,
|
dictToString_numbering, extract_triple_backtick_text,
|
||||||
countGivenWords, remove_french_accents, detect_keyword, extractTextBetweenCharacter,
|
countGivenWords, remove_french_accents, detect_keyword, extractTextBetweenCharacter,
|
||||||
convertCamelSnakeKebabCase
|
extractTextBetweenString,
|
||||||
|
convertCamelSnakeKebabCase, fitrange, recentElementsIndex, nonRecentElementsIndex
|
||||||
|
|
||||||
using JSON3, DataStructures, Distributions, Random, Dates, UUIDs, MQTTClient, DataFrames
|
using JSON3, DataStructures, Distributions, Random, Dates, UUIDs, MQTTClient, DataFrames
|
||||||
|
|
||||||
@@ -42,6 +43,7 @@ function timedifference(starttime::DateTime, stoptime::DateTime, unit::String)::
|
|||||||
diff = stoptime - starttime
|
diff = stoptime - starttime
|
||||||
unit = lowercase(unit)
|
unit = lowercase(unit)
|
||||||
|
|
||||||
|
# Check the unit and calculate the time difference accordingly
|
||||||
if unit == "milliseconds"
|
if unit == "milliseconds"
|
||||||
return diff.value
|
return diff.value
|
||||||
elseif unit == "seconds"
|
elseif unit == "seconds"
|
||||||
@@ -306,7 +308,8 @@ function textToDict(text::String, detectKeywords::Vector{String};
|
|||||||
dictKey_ = reverse(dictKey)
|
dictKey_ = reverse(dictKey)
|
||||||
|
|
||||||
# process text from back to front
|
# process text from back to front
|
||||||
for (i,keyword) in enumerate(reverse(kw))
|
rkw = reverse(kw)
|
||||||
|
for (i,keyword) in enumerate(rkw)
|
||||||
# Find the position of the keyword in the text
|
# Find the position of the keyword in the text
|
||||||
keywordidx = findlast(keyword, remainingtext)
|
keywordidx = findlast(keyword, remainingtext)
|
||||||
dKey = dictKey_[i]
|
dKey = dictKey_[i]
|
||||||
@@ -770,6 +773,123 @@ function extract_triple_backtick_text(input::String)::Vector{String}
|
|||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
|
wordwindow(word::String, startindex::Integer)::UnitRange = startindex:startindex + length(word) -1
|
||||||
|
|
||||||
|
function cuttext(range, text)
|
||||||
|
# check whether range is outside text boundary
|
||||||
|
if range.start > length(text) || range.stop > length(text)
|
||||||
|
return nothing
|
||||||
|
else
|
||||||
|
return text[range]
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
"""
|
||||||
|
detect_keyword(keywords::AbstractVector{String}, text::String; mode::Union{String, Nothing}=nothing, delimiter::AbstractVector=[' ', '\n', '.']) -> Dict{String, Integer}
|
||||||
|
|
||||||
|
Detects and counts occurrences of multiple keywords in the text in different case variations (lowercase, uppercase first letter, or all uppercase).
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `keywords::AbstractVector{String}` Vector of keywords to search for
|
||||||
|
- `text::String` The text to search in
|
||||||
|
|
||||||
|
# Keyword Arguments
|
||||||
|
- `mode::Union{String, Nothing}` When set to "individual", only counts matches that are individual words (default: nothing)
|
||||||
|
- `delimiter::AbstractVector` Characters used to determine word boundaries when mode="individual" (default: [' ', '\n', '.'])
|
||||||
|
|
||||||
|
# Returns
|
||||||
|
- `Dict{String, Integer}` Returns a dictionary mapping each keyword to its count in the text (0 if not found)
|
||||||
|
|
||||||
|
# Examples
|
||||||
|
```jldoctest
|
||||||
|
julia> detect_keyword(["test", "example"], "This is a Test EXAMPLE")
|
||||||
|
Dict{String, Integer}("test" => 1, "example" => 1)
|
||||||
|
|
||||||
|
julia> detect_keyword(["cat"], "cats and category", mode="individual")
|
||||||
|
Dict{String, Integer}("cat" => 0)
|
||||||
|
|
||||||
|
julia> detect_keyword(["error"], "No ERRORS found!")
|
||||||
|
Dict{String, Integer}("error" => 1)
|
||||||
|
```
|
||||||
|
|
||||||
|
# Signature
|
||||||
|
"""
|
||||||
|
# function detect_keyword(keywords::T1, text::String;
|
||||||
|
# mode::Union{String, Nothing}=nothing, delimiter::T2=[' ', '\n', '.']
|
||||||
|
# )::Dict{String, Integer} where {T1<:AbstractVector, T2<:AbstractVector}
|
||||||
|
# # Initialize dictionary to store keyword counts
|
||||||
|
# kwdict = Dict{String, Integer}()
|
||||||
|
# for i in keywords
|
||||||
|
# kwdict[i] = 0
|
||||||
|
# end
|
||||||
|
|
||||||
|
# startindex = 1
|
||||||
|
# # Iterate through each keyword and search for matches in text
|
||||||
|
# for kw in keywords
|
||||||
|
# # Check each possible starting position in the text
|
||||||
|
# for startindex in 1:1:length(text)
|
||||||
|
# # Get the window range for current keyword at current position
|
||||||
|
# wordwindows = wordwindow(kw, startindex)
|
||||||
|
# # Extract the text slice for comparison
|
||||||
|
# cuttexts = cuttext(wordwindows, text)
|
||||||
|
# if cuttexts !== nothing
|
||||||
|
# # Try to detect keyword in current text slice
|
||||||
|
# detected_kw = detect_keyword(kw, cuttexts)
|
||||||
|
# if detected_kw !== nothing && mode === nothing
|
||||||
|
# # Increment count if keyword found and no mode restrictions
|
||||||
|
# kwdict[kw] +=1
|
||||||
|
# elseif detected_kw !== nothing && mode === "individual"
|
||||||
|
# # For individual word mode, check word boundaries
|
||||||
|
# # Check if character before keyword is a delimiter or start of text
|
||||||
|
# checkbefore =
|
||||||
|
# if wordwindows.start > 1 &&
|
||||||
|
# text[wordwindows.start-1] ∈ delimiter
|
||||||
|
# true
|
||||||
|
# elseif wordwindows.start == 1
|
||||||
|
# true
|
||||||
|
# else
|
||||||
|
# false
|
||||||
|
# end
|
||||||
|
|
||||||
|
# # Check if character after keyword is a delimiter or end of text
|
||||||
|
# checkafter =
|
||||||
|
# if wordwindows.stop < length(text) &&
|
||||||
|
# text[wordwindows.stop+1] ∈ delimiter
|
||||||
|
# true
|
||||||
|
# elseif wordwindows.stop == length(text)
|
||||||
|
# true
|
||||||
|
# else
|
||||||
|
# false
|
||||||
|
# end
|
||||||
|
# # Only count keyword if it's a complete word
|
||||||
|
# if checkbefore && checkafter
|
||||||
|
# kwdict[kw] +=1
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
# end
|
||||||
|
# return kwdict
|
||||||
|
# end
|
||||||
|
|
||||||
|
|
||||||
|
function detect_keyword(keywords::T, text::String)::Dict{String, Integer} where {T<:AbstractVector}
|
||||||
|
kw = Dict{String, Integer}()
|
||||||
|
splittext = string.(split(text, " "))
|
||||||
|
# use for loop and detect_keyword function to get the exact variation of each keyword in the text then push to kw list
|
||||||
|
for keyword in keywords
|
||||||
|
ws = detect_keyword.(keyword, splittext)
|
||||||
|
total = sum(issomething.(ws))
|
||||||
|
if total != 0
|
||||||
|
kw[keyword] = total
|
||||||
|
else
|
||||||
|
kw[keyword] = 0
|
||||||
|
end
|
||||||
|
end
|
||||||
|
return kw
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
detect_keyword(keyword::String, text::String) -> Union{Nothing, String}
|
detect_keyword(keyword::String, text::String) -> Union{Nothing, String}
|
||||||
|
|
||||||
@@ -924,9 +1044,11 @@ Extracts and returns the text that is enclosed between two specified characters
|
|||||||
|
|
||||||
# Examples
|
# Examples
|
||||||
```jldoctest
|
```jldoctest
|
||||||
julia> text = "Hello [World]!"
|
julia> text = "Hello [World]! [Yay]"
|
||||||
julia> extracted_text = extractTextBetweenCharacter(text, '[', ']')
|
julia> extracted_text = extractTextBetweenCharacter(text, '[', ']')
|
||||||
println(extracted_text) # Output: "World"
|
2-element Vector{Any}:
|
||||||
|
"World"
|
||||||
|
"Yay"
|
||||||
```
|
```
|
||||||
"""
|
"""
|
||||||
function extractTextBetweenCharacter(text::String, startchar::Char, endchar::Char)
|
function extractTextBetweenCharacter(text::String, startchar::Char, endchar::Char)
|
||||||
@@ -950,6 +1072,29 @@ function extractTextBetweenCharacter(text::String, startchar::Char, endchar::Cha
|
|||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
|
function extractTextBetweenString(text::String, startstr::String, endstr::String)
|
||||||
|
# check whether startstr is in the text or not
|
||||||
|
isStartStr = split(text, startstr)
|
||||||
|
if length(isStartStr) > 2
|
||||||
|
return (success=false, error="There are more than one occurrences of the start string '$startstr' in the text. Text must has only one start string", errorcode=2, result=nothing)
|
||||||
|
elseif length(isStartStr) == 1
|
||||||
|
return (success=false, error="There are no start string '$startstr' in the text. Text must has only one start string", errorcode=1, result=nothing)
|
||||||
|
end
|
||||||
|
|
||||||
|
# check whether endstr is in the text or not
|
||||||
|
isEndStr = split(text, endstr)
|
||||||
|
if length(isEndStr) > 2
|
||||||
|
return (success=false, error="There are more than one occurrences of the end string '$endstr' in the text. Text must has only one end string", errorcode=3, result=nothing)
|
||||||
|
elseif length(isStartStr) == 1
|
||||||
|
return (success=false, error="There are no end string '$endstr' in the text. Text must has only one end string", errorcode=4, result=nothing)
|
||||||
|
end
|
||||||
|
|
||||||
|
s = string(split(isStartStr[2], endstr)[1])
|
||||||
|
|
||||||
|
return (success=true, error=nothing, errorcode=0, text=s)
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Determines if the given string follows camel case naming convention.
|
Determines if the given string follows camel case naming convention.
|
||||||
|
|
||||||
@@ -1075,6 +1220,140 @@ function convertCamelSnakeKebabCase(text::T, tocase::Symbol)::String where {T<:A
|
|||||||
end
|
end
|
||||||
|
|
||||||
|
|
||||||
|
""" Check if a value is not `nothing`.
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `x`: The value to check
|
||||||
|
|
||||||
|
# Returns
|
||||||
|
- `Bool`: `true` if `x` is not `nothing`, `false` otherwise
|
||||||
|
|
||||||
|
# Examples
|
||||||
|
```jldoctest
|
||||||
|
julia> issomething(1)
|
||||||
|
true
|
||||||
|
julia> issomething(nothing)
|
||||||
|
false
|
||||||
|
julia> issomething("test")
|
||||||
|
true
|
||||||
|
````
|
||||||
|
"""
|
||||||
|
function issomething(x)
|
||||||
|
return x === nothing ? false : true
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
""" Adjust a given range to fit within the bounds of a vector's length.
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `v::T1`
|
||||||
|
the input vector to check against
|
||||||
|
- `range::UnitRange`
|
||||||
|
the original range to be adjusted
|
||||||
|
|
||||||
|
# Return
|
||||||
|
- `adjusted_range::UnitRange`
|
||||||
|
a range that is constrained to the vector's length, preventing out-of-bounds indexing
|
||||||
|
|
||||||
|
# Example
|
||||||
|
|
||||||
|
julia> v = [1, 2, 3, 4, 5]
|
||||||
|
julia> fitrange(v, 3:10)
|
||||||
|
3:5
|
||||||
|
|
||||||
|
"""
|
||||||
|
function fitrange(v::T1, range::UnitRange) where {T1<:AbstractVector}
|
||||||
|
totalelements = length(v)
|
||||||
|
|
||||||
|
startind =
|
||||||
|
# check if user put start range greater than total event
|
||||||
|
if range.start > totalelements
|
||||||
|
totalelements
|
||||||
|
else
|
||||||
|
range.start
|
||||||
|
end
|
||||||
|
|
||||||
|
stopind =
|
||||||
|
if range.stop > totalelements
|
||||||
|
totalelements
|
||||||
|
else
|
||||||
|
range.stop
|
||||||
|
end
|
||||||
|
|
||||||
|
return startind:stopind
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
""" Find a unit range for a vector given a number of the most recent elements of interest.
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `vectorLength::Integer`
|
||||||
|
the length of the vector to generate range from
|
||||||
|
- `n::Integer`
|
||||||
|
the number of most recent elements to include in range
|
||||||
|
|
||||||
|
|
||||||
|
# Return
|
||||||
|
- `UnitRange`
|
||||||
|
a range representing the n most recent elements of a vector with length vectorLength
|
||||||
|
|
||||||
|
# Example
|
||||||
|
```jldoctest
|
||||||
|
julia> a = [1, 2, 3, 4, 5]
|
||||||
|
julia> recentElementsIndex(length(a), 3)
|
||||||
|
3:5
|
||||||
|
julia> recentElementsIndex(length(a), 0)
|
||||||
|
5:5
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
function recentElementsIndex(vectorlength::Integer, n::Integer; includelatest::Bool=false)
|
||||||
|
if n == 0
|
||||||
|
error("n must be greater than 0")
|
||||||
|
end
|
||||||
|
|
||||||
|
if includelatest
|
||||||
|
start = max(1, vectorlength - n + 1)
|
||||||
|
return start:vectorlength
|
||||||
|
else
|
||||||
|
startind = max(1, vectorlength - n)
|
||||||
|
endind = vectorlength -1
|
||||||
|
return startind:endind
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
""" Find a unit range for a vector excluding the most recent elements.
|
||||||
|
|
||||||
|
# Arguments
|
||||||
|
- `vectorlength::Integer`
|
||||||
|
the length of the vector to generate range from
|
||||||
|
- `n::Integer`
|
||||||
|
the number of most recent elements to exclude from range
|
||||||
|
|
||||||
|
# Return
|
||||||
|
- `UnitRange`
|
||||||
|
a range representing the elements of the vector excluding the last `n` elements
|
||||||
|
|
||||||
|
# Example
|
||||||
|
```jldoctest
|
||||||
|
julia> a = [1, 2, 3, 4, 5]
|
||||||
|
julia> nonRecentElementsIndex(length(a), 3)
|
||||||
|
1:2
|
||||||
|
julia> nonRecentElementsIndex(length(a), 1)
|
||||||
|
1:4
|
||||||
|
julia> nonRecentElementsIndex(length(a), 0)
|
||||||
|
1:5
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
function nonRecentElementsIndex(vectorlength::Integer, n::Integer)
|
||||||
|
if n < 0
|
||||||
|
error("n must be non-negative")
|
||||||
|
end
|
||||||
|
if n > vectorlength
|
||||||
|
return 1:0 # empty range
|
||||||
|
end
|
||||||
|
return 1:(vectorlength-n)
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
3
test/.vscode/settings.json
vendored
Normal file
3
test/.vscode/settings.json
vendored
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
{
|
||||||
|
"julia.environmentPath": "/appfolder/app/dev/GeneralUtils/test"
|
||||||
|
}
|
||||||
41
test/Manifest.toml
Normal file
41
test/Manifest.toml
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
# This file is machine-generated - editing it directly is not advised
|
||||||
|
|
||||||
|
julia_version = "1.11.4"
|
||||||
|
manifest_format = "2.0"
|
||||||
|
project_hash = "71d91126b5a1fb1020e1098d9d492de2a4438fd2"
|
||||||
|
|
||||||
|
[[deps.Base64]]
|
||||||
|
uuid = "2a0f44e3-6c83-55bd-87e4-b1978d98bd5f"
|
||||||
|
version = "1.11.0"
|
||||||
|
|
||||||
|
[[deps.InteractiveUtils]]
|
||||||
|
deps = ["Markdown"]
|
||||||
|
uuid = "b77e0a4c-d291-57a0-90e8-8db25a27a240"
|
||||||
|
version = "1.11.0"
|
||||||
|
|
||||||
|
[[deps.Logging]]
|
||||||
|
uuid = "56ddb016-857b-54e1-b83d-db4d58db5568"
|
||||||
|
version = "1.11.0"
|
||||||
|
|
||||||
|
[[deps.Markdown]]
|
||||||
|
deps = ["Base64"]
|
||||||
|
uuid = "d6f4376e-aef5-505a-96c1-9c027394607a"
|
||||||
|
version = "1.11.0"
|
||||||
|
|
||||||
|
[[deps.Random]]
|
||||||
|
deps = ["SHA"]
|
||||||
|
uuid = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
|
||||||
|
version = "1.11.0"
|
||||||
|
|
||||||
|
[[deps.SHA]]
|
||||||
|
uuid = "ea8e919c-243c-51af-8825-aaa63cd721ce"
|
||||||
|
version = "0.7.0"
|
||||||
|
|
||||||
|
[[deps.Serialization]]
|
||||||
|
uuid = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
|
||||||
|
version = "1.11.0"
|
||||||
|
|
||||||
|
[[deps.Test]]
|
||||||
|
deps = ["InteractiveUtils", "Logging", "Random", "Serialization"]
|
||||||
|
uuid = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
|
||||||
|
version = "1.11.0"
|
||||||
2
test/Project.toml
Normal file
2
test/Project.toml
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
[deps]
|
||||||
|
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
|
||||||
7
test/etc.jl
Normal file
7
test/etc.jl
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
python -> pandas -> dataframe -> csv
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
julia -> DataFrames -> dataframe -> csv
|
||||||
|
|
||||||
|
dict -> dataframe -> csv
|
||||||
@@ -1,44 +0,0 @@
|
|||||||
using Revise
|
|
||||||
using GeneralUtils, MQTTClient, JSON3
|
|
||||||
|
|
||||||
mqttMsgReceiveTopic = ["/receivetopic_1", "/receivetopic_2"]
|
|
||||||
mqttMsgReceiveChannel = (ch1=Channel(8), ch2=Channel(32))
|
|
||||||
keepaliveChannel = Channel(8)
|
|
||||||
function onMsgCallback(topic, payload)
|
|
||||||
jobj = JSON3.read(String(payload))
|
|
||||||
incomingMqttMsg = copy(jobj) # convert json object into julia dictionary recursively
|
|
||||||
|
|
||||||
if occursin("topic_1", topic)
|
|
||||||
put!(mqttMsgReceiveChannel[:ch1], incomingMqttMsg)
|
|
||||||
elseif occursin("topic_2", topic)
|
|
||||||
put!(mqttMsgReceiveChannel[:ch2], incomingMqttMsg)
|
|
||||||
elseif occursin("keepalive", topic)
|
|
||||||
put!(keepaliveChannel, incomingMqttMsg)
|
|
||||||
else
|
|
||||||
println("undefined condition ", @__FILE__, " ", @__LINE__)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
mqttInstance = GeneralUtils.mqttClientInstance_v2(
|
|
||||||
"mqtt.yiem.cc",
|
|
||||||
mqttMsgReceiveTopic,
|
|
||||||
mqttMsgReceiveChannel,
|
|
||||||
keepaliveChannel,
|
|
||||||
onMsgCallback
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
_ = GeneralUtils.checkMqttConnection!(mqttInstance)
|
|
||||||
|
|
||||||
|
|
||||||
println("GeneralUtils test done")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
39
test/runtests.jl
Normal file
39
test/runtests.jl
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
using Test
|
||||||
|
using GeneralUtils: detect_keyword
|
||||||
|
|
||||||
|
@testset "detect_keyword tests" begin
|
||||||
|
@test detect_keyword(["test"], "this is a test") == Dict("test" => 1)
|
||||||
|
|
||||||
|
@test detect_keyword(["hello", "world"], "hello world hello") == Dict("hello" => 2, "world" => 1)
|
||||||
|
|
||||||
|
@test detect_keyword(["cat"], "category") == Dict("cat" => 1)
|
||||||
|
|
||||||
|
@test detect_keyword(["cat"], "category"; mode="individual") == Dict("cat" => 0)
|
||||||
|
|
||||||
|
@test detect_keyword(["dog"], "dogs and cats"; mode="individual", delimiter=[' ']) == Dict("dog" => 0)
|
||||||
|
|
||||||
|
@test detect_keyword(["test"], "test.case"; mode="individual", delimiter=['.']) == Dict("test" => 1)
|
||||||
|
|
||||||
|
@test detect_keyword(["word"], "") == Dict("word" => 0)
|
||||||
|
|
||||||
|
@test detect_keyword(String[], "some text") == Dict{String, Integer}()
|
||||||
|
|
||||||
|
@test detect_keyword(["a", "b"], "a.b\nc"; delimiter=['.', '\n']) == Dict("a" => 1, "b" => 1)
|
||||||
|
|
||||||
|
multiline_text = """
|
||||||
|
first line
|
||||||
|
second line
|
||||||
|
first word
|
||||||
|
"""
|
||||||
|
@test detect_keyword(["first"], multiline_text) == Dict("first" => 2)
|
||||||
|
|
||||||
|
@test detect_keyword(["word"], "word"; mode="individual") == Dict("word" => 1)
|
||||||
|
|
||||||
|
@test detect_keyword(["test"], "testing.test.tester"; mode="individual", delimiter=['.']) == Dict("test" => 1)
|
||||||
|
end
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Reference in New Issue
Block a user