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

Merge branch 'release/v1.4'

Browse source
This commit is contained in:
Cyber MacGeddon 2025-10-11 11:43:26 +01:00
commit c9b1f2f038
4 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
trustgraph-flow/trustgraph/query/doc_embeddings/qdrant/service.py
View file

@ -61,6 +61,10 @@ class Processor(DocumentEmbeddingsQueryService):
"""Check if collection exists (no implicit creation)"""
return self.qdrant.collection_exists(collection)
def collection_exists(self, collection):
"""Check if collection exists (no implicit creation)"""
return self.qdrant.collection_exists(collection)
async def query_document_embeddings(self, msg):
try:

6
trustgraph-flow/trustgraph/query/graph_embeddings/qdrant/service.py
View file

@ -61,6 +61,10 @@ class Processor(GraphEmbeddingsQueryService):
"""Check if collection exists (no implicit creation)"""
return self.qdrant.collection_exists(collection)
def collection_exists(self, collection):
"""Check if collection exists (no implicit creation)"""
return self.qdrant.collection_exists(collection)
def create_value(self, ent):
if ent.startswith("http://") or ent.startswith("https://"):
return Value(value=ent, is_uri=True)
@ -85,8 +89,6 @@ class Processor(GraphEmbeddingsQueryService):
for vec in msg.vectors:
self.ensure_collection_exists(collection, dim)
# Heuristic hack, get (2*limit), so that we have more chance
# of getting (limit) entities
search_result = self.qdrant.query_points(