File tree Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -61,12 +61,10 @@ def get_embedding_models() -> List[ModelResponse]:
61
61
ModelResponse (
62
62
model_id = "text-embedding-ada-002" ,
63
63
name = "Text Embedding Ada 002" ,
64
- available = True
65
64
),
66
65
ModelResponse (
67
66
model_id = "text-embedding-3-large" ,
68
67
name = "Text Embedding 3 Large" ,
69
- available = True
70
68
),
71
69
]
72
70
Original file line number Diff line number Diff line change 40
40
41
41
from app import config
42
42
from app .services .caii .utils import get_caii_access_token
43
+ from app .services .models import OpenAiModelProvider
43
44
from app .services .models .providers import (
44
45
AzureModelProvider ,
45
46
CAIIModelProvider ,
@@ -69,5 +70,11 @@ def get_crewai_llm_object_direct(
69
70
return CrewAILLM (
70
71
model = "bedrock/" + model_name ,
71
72
)
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
+ )
72
79
else :
73
80
raise ValueError ("Model type is not supported." )
You can’t perform that action at this time.
0 commit comments