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

1) FixBug

Browse Source 
2) Add Documents abount TASK/TODO Refactor
This commit is contained in:
Liu Zhicong
2024-01-06 13:08:41 -08:00
parent 3663f140fc
commit c975ba2053
22 changed files with 398 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/component/openai_node/open_ai_node.py
+1 -1
View File
@@ -223,7 +223,7 @@ class OpenAI_ComputeNode(ComputeNode):
max_tokens=result_token,
)
else:
logger.info(f"call openai {mode_name} prompts: \n\t {prompts} \nfunctions: \n\t{json.dumps(llm_inner_functions)}")
logger.info(f"call openai {mode_name} prompts: \n\t {prompts} \nfunctions: \n\t{json.dumps(llm_inner_functions,ensure_ascii=False)}")
resp = await client.chat.completions.create(model=mode_name,
messages=prompts,
response_format = response_format,
src/component/openai_node/whisper_node.py
+4 -4
View File
@@ -156,11 +156,11 @@ class WhisperComputeNode(ComputeNode):
result.result_str = text
result.result = text
elif response_format == "json":
result.result_str = json.dumps({"text": text})
result.result_str = json.dumps({"text": text},ensure_ascii=False)
resp.text = text
result.result = resp
elif response_format == "verbose_json":
result.result_str = json.dumps({"text": text, "segments": results})
result.result_str = json.dumps({"text": text, "segments": results},ensure_ascii=False)
latest_resp.text = text
latest_resp.segments = results
result.result = latest_resp
@@ -190,9 +190,9 @@ class WhisperComputeNode(ComputeNode):
result.set_from_task(task)
result.worker_id = self.node_id
if response_format == "json":
result.result_str = json.dumps({"text": resp.text})
result.result_str = json.dumps({"text": resp.text},ensure_ascii=False)
elif response_format == "verbose_json":
result.result_str = json.dumps({"text": resp.text, "segments": resp.segments})
result.result_str = json.dumps({"text": resp.text, "segments": resp.segments},ensure_ascii=False)
elif response_format == "srt" or response_format == "vtt" or response_format == "text":
result.result_str = resp
else: