Refactor requirements and code formatting
- Simplified requirements.txt by removing specific version constraints - Added comment about extra library installation using pipmaster - Improved code formatting in base.py, operate.py, and postgres_impl.py - Cleaned up SQL templates and query method signatures with consistent formatting
This commit is contained in:
@@ -893,7 +893,9 @@ async def mix_kg_vector_query(
|
||||
# Reduce top_k for vector search in hybrid mode since we have structured information from KG
|
||||
mix_topk = min(10, query_param.top_k)
|
||||
# TODO: add ids to the query
|
||||
results = await chunks_vdb.query(augmented_query, top_k=mix_topk, ids = query_param.ids)
|
||||
results = await chunks_vdb.query(
|
||||
augmented_query, top_k=mix_topk, ids=query_param.ids
|
||||
)
|
||||
if not results:
|
||||
return None
|
||||
|
||||
@@ -1102,7 +1104,9 @@ async def _get_node_data(
|
||||
f"Query nodes: {query}, top_k: {query_param.top_k}, cosine: {entities_vdb.cosine_better_than_threshold}"
|
||||
)
|
||||
|
||||
results = await entities_vdb.query(query, top_k=query_param.top_k, ids = query_param.ids)
|
||||
results = await entities_vdb.query(
|
||||
query, top_k=query_param.top_k, ids=query_param.ids
|
||||
)
|
||||
|
||||
if not len(results):
|
||||
return "", "", ""
|
||||
@@ -1357,7 +1361,9 @@ async def _get_edge_data(
|
||||
f"Query edges: {keywords}, top_k: {query_param.top_k}, cosine: {relationships_vdb.cosine_better_than_threshold}"
|
||||
)
|
||||
|
||||
results = await relationships_vdb.query(keywords, top_k = query_param.top_k, ids = query_param.ids)
|
||||
results = await relationships_vdb.query(
|
||||
keywords, top_k=query_param.top_k, ids=query_param.ids
|
||||
)
|
||||
|
||||
if not len(results):
|
||||
return "", "", ""
|
||||
@@ -1606,7 +1612,9 @@ async def naive_query(
|
||||
if cached_response is not None:
|
||||
return cached_response
|
||||
|
||||
results = await chunks_vdb.query(query, top_k=query_param.top_k, ids = query_param.ids)
|
||||
results = await chunks_vdb.query(
|
||||
query, top_k=query_param.top_k, ids=query_param.ids
|
||||
)
|
||||
if not len(results):
|
||||
return PROMPTS["fail_response"]
|
||||
|
||||
|
Reference in New Issue
Block a user