gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add openai text embedding compute task

Browse Source
This commit is contained in:
tsukasa
2023-08-31 16:32:20 +08:00
parent 071452fe19
commit 0eebdf61d0
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/compute_task.py
-1
View File
@@ -63,7 +63,6 @@ class ComputeTask:
self.params["model_name"] = model_name self.params["model_name"] = model_name
else: else:
self.params["model_name"] = "text-embedding-ada-002" self.params["model_name"] = "text-embedding-ada-002"
self.params["input"] = input self.params["input"] = input
def display(self) -> str: def display(self) -> str:
src/aios_kernel/knowledge/knowledge_base.py
+2
View File
@@ -8,3 +8,5 @@ class KnowledgeBase:
async def query(self, prompt: AgentPrompt) -> AgentPrompt: async def query(self, prompt: AgentPrompt) -> AgentPrompt:
pass pass