diff --git a/trustgraph/embeddings/hf/hf.py b/trustgraph/embeddings/hf/hf.py index 33890483..671ba879 100755 --- a/trustgraph/embeddings/hf/hf.py +++ b/trustgraph/embeddings/hf/hf.py @@ -1,7 +1,7 @@ """ -Simple LLM service, performs text prompt completion using an Ollama service. -Input is prompt, output is response. +Embeddings service, applies an embeddings model selected from HuggingFace. +Input is text, output is embeddings vector. """ import pulsar diff --git a/trustgraph/embeddings/vectorize/vectorize.py b/trustgraph/embeddings/vectorize/vectorize.py index b362470d..99ee6013 100755 --- a/trustgraph/embeddings/vectorize/vectorize.py +++ b/trustgraph/embeddings/vectorize/vectorize.py @@ -1,7 +1,7 @@ """ -Vectorizer, applies an embedding algorithm to a chunk. Input is a chunk, -output is chunk and vectors. +Vectorizer, calls the embeddings service to get embeddings for a chunk. +Input is text chunk, output is chunk and vectors. """ import pulsar diff --git a/trustgraph/graph/cassandra_write/write.py b/trustgraph/graph/cassandra_write/write.py index 1b56b5ab..f8603a63 100755 --- a/trustgraph/graph/cassandra_write/write.py +++ b/trustgraph/graph/cassandra_write/write.py @@ -1,7 +1,6 @@ """ -Simple decoder, accepts PDF documents on input, outputs pages from the -PDF document as text as separate output objects. +Graph writer. Input is graph edge. Writes edges to Cassandra graph. """ import pulsar diff --git a/trustgraph/kg/extract_definitions/extract.py b/trustgraph/kg/extract_definitions/extract.py index 369aef59..513fd726 100755 --- a/trustgraph/kg/extract_definitions/extract.py +++ b/trustgraph/kg/extract_definitions/extract.py @@ -1,7 +1,7 @@ """ -Simple decoder, accepts PDF documents on input, outputs pages from the -PDF document as text as separate output objects. +Simple decoder, accepts vector+text chunks input, applies entity analysis to +get entity definitions which are output as graph edges. """ import pulsar diff --git a/trustgraph/kg/extract_relationships/extract.py b/trustgraph/kg/extract_relationships/extract.py index a47c3b6e..cd7d1aa1 100755 --- a/trustgraph/kg/extract_relationships/extract.py +++ b/trustgraph/kg/extract_relationships/extract.py @@ -1,7 +1,8 @@ """ -Simple decoder, accepts PDF documents on input, outputs pages from the -PDF document as text as separate output objects. +Simple decoder, accepts vector+text chunks input, applies entity +relationship analysis to get entity relationship edges which are output as +graph edges. """ import pulsar diff --git a/trustgraph/llm/azure_text/llm.py b/trustgraph/llm/azure_text/llm.py index c4f47b6a..26030920 100755 --- a/trustgraph/llm/azure_text/llm.py +++ b/trustgraph/llm/azure_text/llm.py @@ -1,7 +1,7 @@ """ -Simple LLM service, performs text prompt completion using an Ollama service. -Input is prompt, output is response. +Simple LLM service, performs text prompt completion using the Azure +serverless endpoint service. Input is prompt, output is response. """ import pulsar diff --git a/trustgraph/llm/vertexai_text/llm.py b/trustgraph/llm/vertexai_text/llm.py index 44b55e62..d108b6da 100755 --- a/trustgraph/llm/vertexai_text/llm.py +++ b/trustgraph/llm/vertexai_text/llm.py @@ -1,7 +1,7 @@ """ -Simple LLM service, performs text prompt completion using an Ollama service. -Input is prompt, output is response. +Simple LLM service, performs text prompt completion using VertexAI on +Google Cloud. Input is prompt, output is response. """ import pulsar diff --git a/trustgraph/vector/milvus_write/write.py b/trustgraph/vector/milvus_write/write.py index 6a5700d9..67bdcd62 100755 --- a/trustgraph/vector/milvus_write/write.py +++ b/trustgraph/vector/milvus_write/write.py @@ -1,7 +1,6 @@ """ -Simple decoder, accepts PDF documents on input, outputs pages from the -PDF document as text as separate output objects. +Accepts entity/vector pairs and writes them to a Milvus store. """ import pulsar