update
This commit is contained in:
+5
-8
@@ -5,13 +5,10 @@ export get_embedding_nats
|
||||
using GeneralUtils, msghandler
|
||||
|
||||
|
||||
"""
|
||||
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.
|
||||
""" Fetch embeddings for a list of texts from a NATS-based embedding service.
|
||||
|
||||
# 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
|
||||
- `sendto_topic::String`: NATS topic to send the embedding request to
|
||||
- `fileserver_url::String`: URL of the file server for additional data
|
||||
@@ -21,16 +18,16 @@ using GeneralUtils, msghandler
|
||||
|
||||
# Example
|
||||
```julia
|
||||
nats_server_url = "nats://localhost:4222"
|
||||
nats_conn_str = "nats://username:password@yiem.cc:4222"
|
||||
texts = ["Hello world", "Another text"]
|
||||
sendto_topic = "embedding.service"
|
||||
fileserver_url = "http://localhost:8080"
|
||||
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)
|
||||
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)
|
||||
end
|
||||
|
||||
|
||||
Reference in New Issue
Block a user