This commit is contained in:
2026-07-24 21:40:12 +07:00
parent f4bcd2231d
commit 84f3dae629
+5 -8
View File
@@ -5,13 +5,10 @@ export get_embedding_nats
using GeneralUtils, msghandler using GeneralUtils, msghandler
""" """ Fetch embeddings for a list of texts from a NATS-based embedding service.
get_embedding_nats(nats_server_url::String, texts::Vector{String}, sendto_topic::String, fileserver_url::String)
Fetch embeddings for a list of texts from a NATS-based embedding service.
# Arguments # Arguments
- `nats_server_url::String`: NATS server connection URL - `nats_conn_str::String`: NATS server connection URL
- `texts::Vector{String}`: List of text strings to generate embeddings for - `texts::Vector{String}`: List of text strings to generate embeddings for
- `sendto_topic::String`: NATS topic to send the embedding request to - `sendto_topic::String`: NATS topic to send the embedding request to
- `fileserver_url::String`: URL of the file server for additional data - `fileserver_url::String`: URL of the file server for additional data
@@ -21,16 +18,16 @@ using GeneralUtils, msghandler
# Example # Example
```julia ```julia
nats_server_url = "nats://localhost:4222" nats_conn_str = "nats://username:password@yiem.cc:4222"
texts = ["Hello world", "Another text"] texts = ["Hello world", "Another text"]
sendto_topic = "embedding.service" sendto_topic = "embedding.service"
fileserver_url = "http://localhost:8080" fileserver_url = "http://localhost:8080"
embeddings = get_embedding_nats(nats_server_url, texts, sendto_topic, fileserver_url) embeddings = get_embedding_nats(nats_server_url, texts, sendto_topic, fileserver_url)
``` ```
""" """
function get_embedding_nats(nats_server_url::String, texts::Vector{String}, sendto_topic::String, function get_embedding_nats(nats_conn_str::String, texts::Vector{String}, sendto_topic::String,
fileserver_url::String) fileserver_url::String)
nats_conn = NATS.connect(nats_server_url) nats_conn = NATS.connect(nats_conn_str)
return get_embedding_nats(nats_conn, texts, sendto_topic, fileserver_url) return get_embedding_nats(nats_conn, texts, sendto_topic, fileserver_url)
end end