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"})
|
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):
|
||||||
|
|||||||
Reference in New Issue
Block a user