diff --git a/src/ai_powered/decorators.py b/src/ai_powered/decorators.py index 895cb12..7abb3c3 100644 --- a/src/ai_powered/decorators.py +++ b/src/ai_powered/decorators.py @@ -8,8 +8,8 @@ from ai_powered.llm_adapter.generic_adapter import GenericFunctionSimulator from ai_powered.llm_adapter.known_models import complete_model_config from ai_powered.schema_deref import deref -from .constants import DEBUG, OPENAI_API_KEY, OPENAI_BASE_URL, OPENAI_MODEL_NAME, SYSTEM_PROMPT, SYSTEM_PROMPT_JSON_SYNTAX, SYSTEM_PROMPT_RETURN_SCHEMA -from .colors import gray, green +from ai_powered.constants import DEBUG, OPENAI_API_KEY, OPENAI_BASE_URL, OPENAI_MODEL_NAME, SYSTEM_PROMPT, SYSTEM_PROMPT_JSON_SYNTAX, SYSTEM_PROMPT_RETURN_SCHEMA +from ai_powered.colors import gray, green import inspect A = TypeVar("A") diff --git a/src/ai_powered/llm_adapter/generic_adapter.py b/src/ai_powered/llm_adapter/generic_adapter.py index 08c0bf7..6a603ee 100644 --- a/src/ai_powered/llm_adapter/generic_adapter.py +++ b/src/ai_powered/llm_adapter/generic_adapter.py @@ -1,10 +1,10 @@ +from typing import Any, Callable, Set from dataclasses import dataclass import re -from typing import Any, Callable, Set +import openai from ai_powered.colors import green, yellow from ai_powered.constants import DEBUG -from .definitions import FunctionSimulator, ModelFeature -import openai +from ai_powered.llm_adapter.definitions import FunctionSimulator, ModelFeature def _return_schema_wrapper(return_schema: dict[str, Any]) -> dict[str, Any]: return {