Commit f0cf14d4 authored by Your Name's avatar Your Name

Restructure: Move litellm to codai.openai.litellm

- Rename codai/litellm_backend.py to codai/openai/litellm.py
- Create codai/openai/__init__.py
- Update imports in coderai and codai/__init__.py
parent 39f8696e
...@@ -16,21 +16,13 @@ from .models.parser import ( ...@@ -16,21 +16,13 @@ from .models.parser import (
from .models.templates import AgenticTemplateManager from .models.templates import AgenticTemplateManager
# LiteLLM backend (requires litellm package) # OpenAI-compatible backends
try: from .openai.litellm import (
from .litellm_backend import ( LiteLLMBackend,
LiteLLMBackend, get_litellm_backend,
get_litellm_backend, set_litellm_backend,
set_litellm_backend, LITELLM_AVAILABLE,
LITELLM_AVAILABLE, )
)
_LITELLM_IMPORT_ERROR = None
except ImportError as e:
_LITELLM_IMPORT_ERROR = str(e)
LiteLLMBackend = None
get_litellm_backend = None
set_litellm_backend = None
LITELLM_AVAILABLE = False
__all__ = [ __all__ = [
'ModelParserDispatcher', 'ModelParserDispatcher',
......
# codai.openai - OpenAI-compatible API implementations
from .litellm import (
LiteLLMBackend,
get_litellm_backend,
set_litellm_backend,
LITELLM_AVAILABLE,
)
__all__ = [
'LiteLLMBackend',
'get_litellm_backend',
'set_litellm_backend',
'LITELLM_AVAILABLE',
]
...@@ -5175,7 +5175,7 @@ async def chat_completions(request: ChatCompletionRequest): ...@@ -5175,7 +5175,7 @@ async def chat_completions(request: ChatCompletionRequest):
if parser_type == 'litellm': if parser_type == 'litellm':
# Use LiteLLM backend # Use LiteLLM backend
from codai.litellm_backend import get_litellm_backend, LITELLM_AVAILABLE from codai.openai.litellm import get_litellm_backend, LITELLM_AVAILABLE
if not LITELLM_AVAILABLE: if not LITELLM_AVAILABLE:
raise HTTPException( raise HTTPException(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment