From 03de92eec05b7078a47a73922c93366e535bd96c Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 25 Dec 2023 23:01:05 +0530
Subject: [PATCH] (feat) ollama/chat

---
 litellm/llms/ollama_chat.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/litellm/llms/ollama_chat.py b/litellm/llms/ollama_chat.py
index b01c5a7b0b0..b844e285853 100644
--- a/litellm/llms/ollama_chat.py
+++ b/litellm/llms/ollama_chat.py
@@ -125,7 +125,7 @@ class OllamaConfig:
 def get_ollama_response(
     api_base="http://localhost:11434",
     model="llama2",
-    prompt="Why is the sky blue?",
+    messages=None,
     optional_params=None,
     logging_obj=None,
     acompletion: bool = False,
@@ -146,7 +146,7 @@ def get_ollama_response(
             optional_params[k] = v
 
     optional_params["stream"] = optional_params.get("stream", False)
-    data = {"model": model, "prompt": prompt, **optional_params}
+    data = {"model": model, "messages": messages, **optional_params}
     ## LOGGING
     logging_obj.pre_call(
         input=None,
@@ -188,7 +188,7 @@ def get_ollama_response(
 
     ## LOGGING
     logging_obj.post_call(
-        input=prompt,
+        input=messages,
         api_key="",
         original_response=response.text,
         additional_args={