Skip to content

Commit b8a4aed

Browse files
committed
support openai for tool calling
1 parent 4a6d701 commit b8a4aed

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

llm-service/app/services/models/providers/openai.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,12 +61,10 @@ def get_embedding_models() -> List[ModelResponse]:
6161
ModelResponse(
6262
model_id="text-embedding-ada-002",
6363
name="Text Embedding Ada 002",
64-
available=True
6564
),
6665
ModelResponse(
6766
model_id="text-embedding-3-large",
6867
name="Text Embedding 3 Large",
69-
available=True
7068
),
7169
]
7270

llm-service/app/services/query/agents/models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040

4141
from app import config
4242
from app.services.caii.utils import get_caii_access_token
43+
from app.services.models import OpenAiModelProvider
4344
from app.services.models.providers import (
4445
AzureModelProvider,
4546
CAIIModelProvider,
@@ -69,5 +70,11 @@ def get_crewai_llm_object_direct(
6970
return CrewAILLM(
7071
model="bedrock/" + model_name,
7172
)
73+
elif OpenAiModelProvider.is_enabled():
74+
return CrewAILLM(
75+
model="openai/" + model_name,
76+
api_key=config.settings.openai_api_key,
77+
base_url=config.settings.openai_api_base,
78+
)
7279
else:
7380
raise ValueError("Model type is not supported.")

0 commit comments

Comments
 (0)