From 359adfa29aa74f1bcc63a36dff1f8edb843666b9 Mon Sep 17 00:00:00 2001
From: XiaoChenDan <justeought@outlook.com>
Date: Thu, 22 Aug 2024 09:09:14 +0800
Subject: [PATCH] create tool support of Ollama Dotnet

---
 .../ChatCompletion/OllamaChatCompletionService.cs        | 2 ++
 .../ChatCompletion/OllamaChatRequestMessage.cs           | 9 +++++++++
 2 files changed, 11 insertions(+)

diff --git a/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatCompletionService.cs b/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatCompletionService.cs
index d1927f1..d947e7f 100644
--- a/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatCompletionService.cs
+++ b/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatCompletionService.cs
@@ -85,6 +85,8 @@ private IEnumerable<OllamaChatRequestMessage> GetChatMessages(ChatHistory chat)
                 yield return new OllamaChatRequestUserMessage(item.Content!);
             else if (item.Role == AuthorRole.Assistant)
                 yield return new OllamaChatRequestAssistantMessage(item.Content!);
+            else if (item.Role == AuthorRole.Tool)
+                yield return new OllamaChatRequestToolMessage(item.Content!);
         }
     }
 }
\ No newline at end of file
diff --git a/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatRequestMessage.cs b/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatRequestMessage.cs
index 18950f9..3c4da9f 100644
--- a/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatRequestMessage.cs
+++ b/dotnet/Codeblaze.SemanticKernel.Connectors.Ollama/ChatCompletion/OllamaChatRequestMessage.cs
@@ -50,4 +50,13 @@ public OllamaChatRequestAssistantMessage(string content)
             Content = content;
         }
     }
+
+    internal sealed class OllamaChatRequestToolMessage : OllamaChatRequestMessage
+    {
+        public OllamaChatRequestToolMessage(string content)
+        {
+            Role = "tool";
+            Content = content;
+        }
+    }
 }