env set up for remote

This commit is contained in:
2025-01-24 04:41:14 +00:00
parent e1a24aff20
commit d45f0be314
5 changed files with 83 additions and 51 deletions
+3 -1
View File
@@ -36,7 +36,9 @@ class OllamaGenerator:
])
# the deepseek model returns <think> this removes those tabs from the output
return re.sub(r"<think|.\n\r+?|([^;]*)\/think>",'',self.response['message']['content'])
# return re.sub(r"<think|.\n\r+?|([^;]*)\/think>",'',self.response['message']['content'])
return self.response['message']['content']
except Exception as e:
raise Exception(f"Failed to generate markdown: {e}")