Imports and a print
This commit is contained in:
@@ -3,8 +3,6 @@ import re
|
|||||||
import logging
|
import logging
|
||||||
from typing import List, Optional, Callable, TypeVar
|
from typing import List, Optional, Callable, TypeVar
|
||||||
|
|
||||||
from numba.core.transforms import consolidate_multi_exit_withs
|
|
||||||
from numba.cuda import const
|
|
||||||
from openai import AsyncOpenAI
|
from openai import AsyncOpenAI
|
||||||
from openai.types.chat import ChatCompletionMessageParam
|
from openai.types.chat import ChatCompletionMessageParam
|
||||||
|
|
||||||
@@ -76,7 +74,6 @@ class OpenAI(ILLMService):
|
|||||||
return await self._prediction(
|
return await self._prediction(
|
||||||
model, messages, token_count, fields_to_check, temperature, (try_count + 1), check_blacklisted
|
model, messages, token_count, fields_to_check, temperature, (try_count + 1), check_blacklisted
|
||||||
)
|
)
|
||||||
print(result)
|
|
||||||
return json.loads(result)
|
return json.loads(result)
|
||||||
|
|
||||||
async def prediction_override(self, **kwargs):
|
async def prediction_override(self, **kwargs):
|
||||||
|
|||||||
Reference in New Issue
Block a user