Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

udpate k

Browse Source
This commit is contained in:
goulustis
2025-10-17 16:55:37 +08:00
parent a3de7105f0
commit 7f64703952
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lang_agent/rag/simple.py
View File

@@ -24,7 +24,7 @@ class SimpleRagConfig(ToolConfig):
api_key:str = "wrong-key" api_key:str = "wrong-key"
"""api_key for model; for generic text splitting; give a wrong key <-- wrong, MUST have api key""" """api_key for model; for generic text splitting; give a wrong key <-- wrong, MUST have api key"""
folder_path:str = "assets/xiaozhan_emb" folder_path:str = "/home/smith/projects/work/langchain-agent/assets/xiaozhan_emb"
"""path to local database""" """path to local database"""
def __post_init__(self): def __post_init__(self):
@@ -68,7 +68,7 @@ class SimpleRag(LangToolBase):
2. 用户搜索“适合夏天的清爽饮品”，系统会检索并返回相关饮品推荐及其来源信息。 2. 用户搜索“适合夏天的清爽饮品”，系统会检索并返回相关饮品推荐及其来源信息。
""" """
retrieved_docs:List[Document] = self.vec_store.similarity_search(query, retrieved_docs:List[Document] = self.vec_store.similarity_search(query,
k=3) k=10)
serialized = "\n\n".join( serialized = "\n\n".join(
(f"Source: {doc.metadata}\nContent: {doc.page_content}") (f"Source: {doc.metadata}\nContent: {doc.page_content}")
for doc in retrieved_docs for doc in retrieved_docs