Spaces:
Running
Running
Eddie Pick
commited on
Commit
•
ca913e4
1
Parent(s):
7402de3
Bugfix
Browse files- 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('
|
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 |
|