update tests

tests/unit_tests/chains/test_natbot.py

@@ -12,9 +12,9 @@ class FakeLLM(LLM):
     def generate(self, prompt: str, stop: Optional[List[str]] = None) -> List[CompletionOutput]:
         """Return `foo` if longer than 10000 words, else `bar`."""
         if len(prompt) > 10000:
-            return [CompletionOutput("foo")]
+            return [CompletionOutput(text="foo")]
         else:
-            return [CompletionOutput("bar")]
+            return [CompletionOutput(text="bar")]
 
 
 def test_proper_inputs() -> None:

tests/unit_tests/chains/test_react.py

@@ -33,7 +33,7 @@ class FakeListLLM(LLM):
     def generate(self, prompt: str, stop: Optional[List[str]] = None) -> List[CompletionOutput]:
         """Increment counter, and then return response in that index."""
         self.i += 1
-        return [CompletionOutput(self.responses[self.i])]
+        return [CompletionOutput(text=self.responses[self.i])]
 
 
 class FakeDocstore(Docstore):

tests/unit_tests/llms/fake_llm.py

@@ -14,8 +14,8 @@ class FakeLLM(LLM):
     def generate(self, prompt: str, stop: Optional[List[str]] = None) -> List[CompletionOutput]:
         """First try to lookup in queries, else return 'foo' or 'bar'."""
         if self._queries is not None:
-            return self._queries[prompt]
+            return [CompletionOutput(text=self._queries[prompt])]
         if stop is None:
-            return [CompletionOutput("foo")]
+            return [CompletionOutput(text="foo")]
         else:
-            return [CompletionOutput("bar")]
+            return [CompletionOutput(text="bar")]