Eddie Pick commited on
Commit
e5a770a
1 Parent(s): ca913e4
Files changed (1) hide show
  1. 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
- console.log(f"Using model: {chat.model_name}")
106
- console.log(f"Using embedding model: { embedding_model.model}")
 
 
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)