Merge pull request #689 from wwulfric/main

Prompt Content not print unicode char
vanna-ai · Oct 29, 2024 · e640d58 · e640d58
2 parents 29cc557 + 44bd884
commit e640d58
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/vanna/ollama/ollama.py b/src/vanna/ollama/ollama.py
@@ -91,7 +91,7 @@ def submit_prompt(self, prompt, **kwargs) -> str:
       f"model={self.model},\n"
       f"options={self.ollama_options},\n"
       f"keep_alive={self.keep_alive}")
-    self.log(f"Prompt Content:\n{json.dumps(prompt)}")
+    self.log(f"Prompt Content:\n{json.dumps(prompt, ensure_ascii=False)}")
     response_dict = self.ollama_client.chat(model=self.model,
                                             messages=prompt,
                                             stream=False,

diff --git a/src/vanna/remote.py b/src/vanna/remote.py
@@ -62,7 +62,7 @@ def assistant_message(self, message: str) -> any:
 
     def submit_prompt(self, prompt, **kwargs) -> str:
         # JSON-ify the prompt
-        json_prompt = json.dumps(prompt)
+        json_prompt = json.dumps(prompt, ensure_ascii=False)
 
         params = [StringData(data=json_prompt)]