sat_ade/task-6a02e23da6fe2e4ac16acf65-1
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add tools.py

Browse Source
This commit is contained in:
sat_ade committed 2026-06-04 14:30:18 +00:00
1 parent 43724b201e
commit 87932daef8
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
tools.py
+1 -1
View File
@@ -12,7 +12,7 @@ if not client.has_collection(collection_name):
client.create_collection(name=collection_name, vectors_config={"size": 384, "distance": "Cosine"}) client.create_collection(name=collection_name, vectors_config={"size": 384, "distance": "Cosine"})
embeddings = OllamaEmbeddings(model="nomic-embed-text") embeddings = OllamaEmbeddings(model="nomic-embed-text")
text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50) text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=100)
@tool("search_knowledge_base", description="Semantic search in knowledge base") @tool("search_knowledge_base", description="Semantic search in knowledge base")
def search_knowledge_base(query: str, max_results: int = 5): def search_knowledge_base(query: str, max_results: int = 5):