set to string

Josh-XT · Jun 26, 2024 · abc0bd0 · abc0bd0
1 parent 986dc86
commit abc0bd0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agixt/Agent.py b/agixt/Agent.py
@@ -342,7 +342,7 @@ async def inference(self, prompt: str, tokens: int = 0, images: list = []):
         answer = await self.PROVIDER.inference(
             prompt=prompt, tokens=tokens, images=images
         )
-        answer = answer.replace("\_", "_")
+        answer = str(answer).replace("\_", "_")
         if answer.endswith("\n\n"):
             answer = answer[:-2]
         return answer
@@ -355,7 +355,7 @@ async def vision_inference(self, prompt: str, tokens: int = 0, images: list = []
         answer = await self.VISION_PROVIDER.inference(
             prompt=prompt, tokens=tokens, images=images
         )
-        answer = answer.replace("\_", "_")
+        answer = str(answer).replace("\_", "_")
         if answer.endswith("\n\n"):
             answer = answer[:-2]
         return answer