Update ollama_model.py

jagapiou · Aug 9, 2024 · 5d5b652 · 5d5b652
1 parent 5725e10
commit 5d5b652
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/concordia/language_model/ollama_model.py b/concordia/language_model/ollama_model.py
@@ -90,7 +90,8 @@ def sample_text(
         prompt=prompt_with_system_message,
         options={'stop': terminators},
         keep_alive='10m',
-    ).response
+    )
+    result = response.response
 
     if self._measurements is not None:
       self._measurements.publish_datum(