Spaces:
Running
Running
Eddie Pick
commited on
Commit
•
e5a770a
1
Parent(s):
ca913e4
bugfix
Browse files- search_agent.py +4 -2
search_agent.py
CHANGED
@@ -102,8 +102,10 @@ def main(arguments):
|
|
102 |
embedding_model = md.get_embedding_model(embedding_model)
|
103 |
|
104 |
if verbose:
|
105 |
-
|
106 |
-
|
|
|
|
|
107 |
|
108 |
with console.status(f"[bold green]Optimizing query for search: {query}"):
|
109 |
optimize_search_query = wr.optimize_search_query(chat, query)
|
|
|
102 |
embedding_model = md.get_embedding_model(embedding_model)
|
103 |
|
104 |
if verbose:
|
105 |
+
model_name = getattr(chat, 'model_name', None) or getattr(chat, 'model', None) or getattr(chat, 'model_id', None) or str(chat)
|
106 |
+
embedding_model_name = getattr(embedding_model, 'model_name', None) or getattr(embedding_model, 'model', None) or getattr(embedding_model, 'model_id', None) or str(embedding_model)
|
107 |
+
console.log(f"Using model: {model_name}")
|
108 |
+
console.log(f"Using embedding model: {embedding_model_name}")
|
109 |
|
110 |
with console.status(f"[bold green]Optimizing query for search: {query}"):
|
111 |
optimize_search_query = wr.optimize_search_query(chat, query)
|