diff --git a/libs/ollama.py b/libs/ollama.py index 1494c7b..3bdb462 100644 --- a/libs/ollama.py +++ b/libs/ollama.py @@ -67,4 +67,4 @@ def create_prompt_on_openwebui(prompt: str) -> str: # ) # prompt = response["choices"][0]["message"]["content"].strip('"') logging.debug(prompt) - return prompt \ No newline at end of file + return prompt.split(": ")[-1] \ No newline at end of file