|
12 | 12 | from uipath_langchain_client.clients.bedrock.embeddings import UiPathBedrockEmbeddings |
13 | 13 | from uipath_langchain_client.clients.google.chat_models import UiPathChatGoogleGenerativeAI |
14 | 14 | from uipath_langchain_client.clients.google.embeddings import UiPathGoogleGenerativeAIEmbeddings |
15 | | -from uipath_langchain_client.clients.normalized.chat_models import UiPathNormalizedChatModel |
| 15 | +from uipath_langchain_client.clients.normalized.chat_models import UiPathChat |
16 | 16 | from uipath_langchain_client.clients.openai.chat_models import ( |
17 | 17 | UiPathAzureChatOpenAI, |
18 | 18 | UiPathChatOpenAI, |
|
28 | 28 | GPT_MODELS_NON_REASONING_CONFIGS = [ |
29 | 29 | {"model_class": UiPathAzureChatOpenAI}, |
30 | 30 | {"model_class": UiPathAzureChatOpenAI, "model_kwargs": {"use_responses_api": True}}, |
31 | | - # {"model_class": UiPathNormalizedChatModel}, |
| 31 | + # {"model_class": UiPathChat}, |
32 | 32 | ] |
33 | 33 |
|
34 | 34 | GPT_MODELS_WITH_REASONING_CONFIGS = [ |
|
43 | 43 | "verbosity": "low", |
44 | 44 | }, |
45 | 45 | }, |
46 | | - # {"model_class": UiPathNormalizedChatModel, "model_kwargs": {"reasoning_effort": "low"}}, |
| 46 | + # {"model_class": UiPathChat, "model_kwargs": {"reasoning_effort": "low"}}, |
47 | 47 | ] |
48 | 48 |
|
49 | 49 | GEMINI_2_5_CONFIGS = [ |
|
56 | 56 | "model_kwargs": {"thinking_budget": 128, "include_thoughts": True}, |
57 | 57 | }, |
58 | 58 | # { |
59 | | - # "model_class": UiPathNormalizedChatModel, |
| 59 | + # "model_class": UiPathChat, |
60 | 60 | # "model_kwargs": {"thinking_budget": 128, "include_thoughts": True}, |
61 | 61 | # }, |
62 | 62 | ] |
|
71 | 71 | "model_kwargs": {"thinking_level": "low", "include_thoughts": True}, |
72 | 72 | }, |
73 | 73 | # { |
74 | | - # "model_class": UiPathNormalizedChatModel, |
| 74 | + # "model_class": UiPathChat, |
75 | 75 | # "model_kwargs": {"thinking_level": "low", "include_thoughts": True}, |
76 | 76 | # }, |
77 | 77 | ] |
|
101 | 101 | "thinking": {"type": "enabled", "budget_tokens": 1024}, |
102 | 102 | }, |
103 | 103 | }, |
104 | | - # {"model_class": UiPathNormalizedChatModel}, |
| 104 | + # {"model_class": UiPathChat}, |
105 | 105 | ] |
106 | 106 |
|
107 | 107 | CLAUDE_MODELS_AWSBEDROCK_CONFIGS = [ |
|
139 | 139 | "thinking": {"type": "enabled", "budget_tokens": 1024}, |
140 | 140 | }, |
141 | 141 | }, |
142 | | - # {"model_class": UiPathNormalizedChatModel}, |
| 142 | + # {"model_class": UiPathChat}, |
143 | 143 | ] |
144 | 144 |
|
145 | 145 | COMPLETIONS_MODELS_WITH_CONFIGS = { |
|
164 | 164 |
|
165 | 165 |
|
166 | 166 | COMPLETION_CLIENTS_CLASSES = [ |
167 | | - UiPathNormalizedChatModel, |
| 167 | + UiPathChat, |
168 | 168 | UiPathChatOpenAI, |
169 | 169 | UiPathAzureChatOpenAI, |
170 | 170 | # UiPathAzureAIChatCompletionsModel, |
|
175 | 175 | # UiPathChatBedrockConverse, |
176 | 176 | ] |
177 | 177 | EMBEDDINGS_CLIENTS_CLASSES = [ |
178 | | - # UiPathNormalizedEmbeddings, |
| 178 | + # UiPathEmbeddings, |
179 | 179 | UiPathOpenAIEmbeddings, |
180 | 180 | UiPathAzureOpenAIEmbeddings, |
181 | 181 | # UiPathAzureAIEmbeddingsModel, |
@@ -214,11 +214,11 @@ def completions_config( |
214 | 214 | EMBEDDINGS_MODELS_WITH_CONFIGS = { |
215 | 215 | "text-embedding-3-large": [ |
216 | 216 | {"model_class": UiPathAzureOpenAIEmbeddings}, |
217 | | - # {"model_class": UiPathNormalizedEmbeddings}, |
| 217 | + # {"model_class": UiPathEmbeddings}, |
218 | 218 | ], |
219 | 219 | "gemini-embedding-001": [ |
220 | 220 | {"model_class": UiPathGoogleGenerativeAIEmbeddings}, |
221 | | - # {"model_class": UiPathNormalizedEmbeddings}, |
| 221 | + # {"model_class": UiPathEmbeddings}, |
222 | 222 | ], |
223 | 223 | } |
224 | 224 |
|
|
0 commit comments