apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-08 21:02:12 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix vector extraction

Browse source
This commit is contained in:
Cyber MacGeddon 2026-03-08 18:52:07 +00:00
parent 0a2ce47a88
commit aca5ee5653
3 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
trustgraph-flow/trustgraph/embeddings/document_embeddings/embeddings.py
View file

@ -67,8 +67,7 @@ class Processor(FlowProcessor):
)
# vectors[0] is the vector set for the first (only) text
# vectors[0][0] is the first vector in that set
vectors = resp.vectors[0][0] if resp.vectors else []
vectors = resp.vectors[0] if resp.vectors else []
embeds = [
ChunkEmbeddings(

2
trustgraph-flow/trustgraph/embeddings/graph_embeddings/embeddings.py
View file

@ -72,7 +72,7 @@ class Processor(FlowProcessor):
entities = [
EntityEmbeddings(
entity=entity.entity,
vectors=vectors[0], # First vector from the set
vectors=vectors, # Vector set for this entity
chunk_id=entity.chunk_id, # Provenance: source chunk
)
for entity, vectors in zip(v.entities, all_vectors)

2
trustgraph-flow/trustgraph/embeddings/row_embeddings/embeddings.py
View file

@ -216,7 +216,7 @@ class Processor(CollectionConfigHandler, FlowProcessor):
index_name=index_name,
index_value=index_value,
text=text,
vectors=vectors[0] # First vector from the set
vectors=vectors # Vector set for this text
)
)