From 0eebdf61d06eed6a079a1dc3d615ec7e2b411a85 Mon Sep 17 00:00:00 2001
From: tsukasa <puotosssa@live.com>
Date: Thu, 31 Aug 2023 16:32:20 +0800
Subject: [PATCH] Add openai text embedding compute task

---
 src/aios_kernel/compute_task.py             | 1 -
 src/aios_kernel/knowledge/knowledge_base.py | 4 +++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/aios_kernel/compute_task.py b/src/aios_kernel/compute_task.py
index dd875a2..c6053db 100644
--- a/src/aios_kernel/compute_task.py
+++ b/src/aios_kernel/compute_task.py
@@ -63,7 +63,6 @@ class ComputeTask:
             self.params["model_name"] = model_name
         else:
             self.params["model_name"] = "text-embedding-ada-002"
-
         self.params["input"] = input
 
     def display(self) -> str:
diff --git a/src/aios_kernel/knowledge/knowledge_base.py b/src/aios_kernel/knowledge/knowledge_base.py
index 59e4f25..a576280 100644
--- a/src/aios_kernel/knowledge/knowledge_base.py
+++ b/src/aios_kernel/knowledge/knowledge_base.py
@@ -7,4 +7,6 @@ class KnowledgeBase:
         pass
 
     async def query(self, prompt: AgentPrompt) -> AgentPrompt:
-        pass
\ No newline at end of file
+        pass
+
+    
\ No newline at end of file