tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-10 04:57:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: hidden vector content for search param log

Browse Source
This commit is contained in:
arkohut 2024-08-29 21:16:36 +08:00
parent 7fbb05e0cd
commit 2135b11f6c
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
memos/indexing.py
View File

@ -62,7 +62,9 @@ def get_embeddings(texts: List[str]) -> List[List[float]]:
for embedding in response.json()["embeddings"] for embedding in response.json()["embeddings"]
] ]
else: else:
raise Exception(f"Failed to get embeddings: {response.text} {response.status_code}") raise Exception(
f"Failed to get embeddings: {response.text} {response.status_code}"
)
def generate_metadata_text(metadata_entries): def generate_metadata_text(metadata_entries):
@ -310,7 +312,9 @@ def search_entities(
"vector_query": f"embedding:({embedding}, k:{limit})", "vector_query": f"embedding:({embedding}, k:{limit})",
} }
print(json.dumps(search_parameters, indent=2)) search_parameters_to_print = search_parameters.copy()
search_parameters_to_print["vector_query"] = f"embedding:([...], k:{limit})"
print(json.dumps(search_parameters_to_print, indent=2))
search_response = client.multi_search.perform( search_response = client.multi_search.perform(
{"searches": [search_parameters]}, common_search_params {"searches": [search_parameters]}, common_search_params