File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
test/agentchat/contrib/retrievechat Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -54,17 +54,19 @@ def test_retrievechat():
54
54
)
55
55
56
56
sentence_transformer_ef = ef .SentenceTransformerEmbeddingFunction ()
57
+ docs_path = os .path .abspath (os .path .join (os .path .dirname (__file__ ), "../../../../website/docs" ))
57
58
ragproxyagent = RetrieveUserProxyAgent (
58
59
name = "ragproxyagent" ,
59
60
human_input_mode = "NEVER" ,
60
61
max_consecutive_auto_reply = 2 ,
61
62
retrieve_config = {
62
- "docs_path" : "./website/docs" ,
63
+ "docs_path" : docs_path ,
63
64
"chunk_token_size" : 2000 ,
64
65
"model" : config_list [0 ]["model" ],
65
66
"client" : chromadb .PersistentClient (path = "/tmp/chromadb" ),
66
67
"embedding_function" : sentence_transformer_ef ,
67
68
"get_or_create" : True ,
69
+ "overwrite" : True ,
68
70
},
69
71
)
70
72
You can’t perform that action at this time.
0 commit comments