fix: fixes tests
This commit is contained in:
parent
78fb415892
commit
b6be61776a
2 changed files with 2 additions and 1 deletions
|
|
@ -145,7 +145,7 @@ class GraphCompletionRetriever(BaseRetriever):
|
||||||
|
|
||||||
if len(triplets) == 0:
|
if len(triplets) == 0:
|
||||||
logger.warning("Empty context was provided to the completion")
|
logger.warning("Empty context was provided to the completion")
|
||||||
return ""
|
return "", triplets
|
||||||
|
|
||||||
context = await self.resolve_edges_to_text(triplets)
|
context = await self.resolve_edges_to_text(triplets)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -65,6 +65,7 @@ async def test_search(
|
||||||
top_k=10,
|
top_k=10,
|
||||||
node_type=None,
|
node_type=None,
|
||||||
node_name=None,
|
node_name=None,
|
||||||
|
save_interaction=False,
|
||||||
)
|
)
|
||||||
|
|
||||||
# Verify result logging
|
# Verify result logging
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue