add configuable funcname

This commit is contained in:
yichuan520030910320
2025-07-01 05:02:01 +00:00
parent b81b5d0f86
commit 371e3de04e
4 changed files with 973 additions and 1131 deletions

View File

@@ -69,7 +69,7 @@ async def main():
query = "Based on the paper, what are the main techniques LEANN explores to reduce the storage overhead?"
print(f"You: {query}")
chat_response = chat.ask(query, recompute_beighbor_embeddings=True)
chat_response = chat.ask(query, top_k=10, recompute_beighbor_embeddings=True)
print(f"Leann: {chat_response}")
if __name__ == "__main__":