update
This commit is contained in:
+5
-8
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user