Vaishak G Kumar commited on
Commit
98971c8
·
verified ·
1 Parent(s): 1118b3d

Update src/agentics/agents.py

Browse files
Files changed (1) hide show
  1. src/agentics/agents.py +3 -0
src/agentics/agents.py CHANGED
@@ -8,6 +8,9 @@ import chromadb
8
  config_list = autogen.config_list_from_json(
9
  "OAI_CONFIG_LIST.json",
10
  file_location="./src/config/",
 
 
 
11
  )
12
 
13
  print("LLM models: ", [config_list[i]["model"] for i in range(len(config_list))])
 
8
  config_list = autogen.config_list_from_json(
9
  "OAI_CONFIG_LIST.json",
10
  file_location="./src/config/",
11
+ filter_dict={
12
+ "model": ["gpt-3.5-turbo-preview", "gpt-4-preview", "gpt-4-vision-preview", "dall-e-3"],
13
+ },
14
  )
15
 
16
  print("LLM models: ", [config_list[i]["model"] for i in range(len(config_list))])