add image insert/query in knowledge base

This commit is contained in:
tsukasa
2023-09-28 16:20:30 +08:00
parent e0c4eb5849
commit 56c0202552
3 changed files with 47 additions and 20 deletions
+16
View File
@@ -5,6 +5,7 @@ import logging
import asyncio
from asyncio import Queue
from knowledge import ObjectID
from .agent import AgentPrompt
from .compute_node import ComputeNode
from .compute_task import ComputeTask, ComputeTaskState, ComputeTaskResult, ComputeTaskType,ComputeTaskResultCode
@@ -152,6 +153,21 @@ class ComputeKernel:
return "error!"
def image_embedding(self,input:ObjectID,model_name:Optional[str] = None):
task_req = ComputeTask()
task_req.set_image_embedding_params(input,model_name)
self.run(task_req)
return task_req
async def do_image_embedding(self,input:ObjectID,model_name:Optional[str] = None) -> [float]:
task_req = self.image_embedding(input,model_name)
task_result = await self._send_task(task_req)
if task_req.state == ComputeTaskState.DONE:
return task_result.result_str
return "error!"
async def do_text_to_speech(self,
input:str,
language_code:Optional[str] = None,