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

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] is the vector set for the first (only) text
# vectors[0][0] is the first vector in that set vectors = resp.vectors[0] if resp.vectors else []
vectors = resp.vectors[0][0] if resp.vectors else []
embeds = [ embeds = [
ChunkEmbeddings( ChunkEmbeddings(

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

@ -72,7 +72,7 @@ class Processor(FlowProcessor):
entities = [ entities = [
EntityEmbeddings( EntityEmbeddings(
entity=entity.entity, 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 chunk_id=entity.chunk_id, # Provenance: source chunk
) )
for entity, vectors in zip(v.entities, all_vectors) 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_name=index_name,
index_value=index_value, index_value=index_value,
text=text, text=text,
vectors=vectors[0] # First vector from the set vectors=vectors # Vector set for this text
) )
) )