add tools.py
This commit is contained in:
@@ -12,7 +12,7 @@ if not client.has_collection(collection_name):
|
||||
client.create_collection(name=collection_name, vectors_config={"size": 384, "distance": "Cosine"})
|
||||
|
||||
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")
|
||||
def search_knowledge_base(query: str, max_results: int = 5):
|
||||
|
||||
Reference in New Issue
Block a user