Eddie Pick commited on
Commit
ca913e4
1 Parent(s): 7402de3
Files changed (1) hide show
  1. search_agent.py +2 -2
search_agent.py CHANGED
@@ -96,8 +96,8 @@ def main(arguments):
96
 
97
  chat = md.get_model(model, temperature)
98
  if embedding_model.lower() == "same provider as model":
99
- provider = model.split('/')[0]
100
- embedding_model = md.get_embedding_model(f"{provider}/")
101
  else:
102
  embedding_model = md.get_embedding_model(embedding_model)
103
 
 
96
 
97
  chat = md.get_model(model, temperature)
98
  if embedding_model.lower() == "same provider as model":
99
+ provider = model.split(':')[0]
100
+ embedding_model = md.get_embedding_model(f"{provider}")
101
  else:
102
  embedding_model = md.get_embedding_model(embedding_model)
103