Skip to content

Commit c5d51bb

Browse files
committed
Fix retrieval test
1 parent 4095916 commit c5d51bb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/agentchat/contrib/retrievechat/test_retrievechat.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,17 +54,19 @@ def test_retrievechat():
5454
)
5555

5656
sentence_transformer_ef = ef.SentenceTransformerEmbeddingFunction()
57+
docs_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../../website/docs"))
5758
ragproxyagent = RetrieveUserProxyAgent(
5859
name="ragproxyagent",
5960
human_input_mode="NEVER",
6061
max_consecutive_auto_reply=2,
6162
retrieve_config={
62-
"docs_path": "./website/docs",
63+
"docs_path": docs_path,
6364
"chunk_token_size": 2000,
6465
"model": config_list[0]["model"],
6566
"client": chromadb.PersistentClient(path="/tmp/chromadb"),
6667
"embedding_function": sentence_transformer_ef,
6768
"get_or_create": True,
69+
"overwrite": True,
6870
},
6971
)
7072

0 commit comments

Comments
 (0)