Danialebrat commited on
Commit
9023f08
·
1 Parent(s): 86ff876

Fixing bugs and ollama errors

Browse files

eliminating json format for better output

Files changed (1) hide show
  1. Messaging_system/LLM.py +1 -1
Messaging_system/LLM.py CHANGED
@@ -131,7 +131,7 @@ class LLM:
131
  for attempt in range(max_retries):
132
  try:
133
  # Try generating the response
134
- response = self.client.generate(model=self.model, prompt=prompt, format='json')
135
  except Exception as e:
136
  # This catches errors like the connection being forcibly closed
137
  print(f"Error on attempt {attempt + 1}: {e}.")
 
131
  for attempt in range(max_retries):
132
  try:
133
  # Try generating the response
134
+ response = self.client.generate(model=self.model, prompt=prompt)
135
  except Exception as e:
136
  # This catches errors like the connection being forcibly closed
137
  print(f"Error on attempt {attempt + 1}: {e}.")