Fixed level issues

This commit is contained in:
Carlos-Mesquita
2024-11-10 04:21:36 +00:00
parent cf1b676312
commit 6909d75eb6
15 changed files with 101 additions and 84 deletions

View File

@@ -57,7 +57,7 @@ class Heygen(IVideoGeneratorService):
)
async def pool_status(self, video_id: str) -> Task:
async def poll_status(self, video_id: str) -> Task:
response = await self._http_client.get(self._GET_VIDEO_URL, headers=self._get_header, params={
'video_id': video_id
})
@@ -65,7 +65,6 @@ class Heygen(IVideoGeneratorService):
status = response_data["data"]["status"]
error = response_data["data"]["error"]
if status != "completed" and error is None:
self._logger.info(f"Status: {status}")
return Task(

View File

@@ -73,7 +73,7 @@ class OpenAI(ILLMService):
return await self._prediction(
model, messages, token_count, fields_to_check, temperature, (try_count + 1), check_blacklisted
)
print(result)
return json.loads(result)
async def prediction_override(self, **kwargs):