KevinHuSh commited on
Commit
ef15d2d
·
1 Parent(s): 7ccbb61

fix ollama issuet push (#486)

Browse files

### What problem does this PR solve?

#477

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Files changed (1) hide show
  1. rag/llm/chat_model.py +1 -1
rag/llm/chat_model.py CHANGED
@@ -153,7 +153,7 @@ class OllamaChat(Base):
153
  options=options
154
  )
155
  ans = response["message"]["content"].strip()
156
- return ans, response["eval_count"] + response["prompt_eval_count"]
157
  except Exception as e:
158
  return "**ERROR**: " + str(e), 0
159
 
 
153
  options=options
154
  )
155
  ans = response["message"]["content"].strip()
156
+ return ans, response["eval_count"] + response.get("prompt_eval_count", 0)
157
  except Exception as e:
158
  return "**ERROR**: " + str(e), 0
159