updated UiPathChat to normalized client

Author: cosminacho

samples/oauth-external-apps-agent/main.py

@@ -14,7 +14,7 @@
 from pydantic import BaseModel
 from uipath.platform import UiPath
 
-from uipath_langchain.chat import UiPathChatAnthropic
+from uipath_langchain.chat import UiPathChat
 
 dotenv.load_dotenv()
 
@@ -63,9 +63,7 @@ async def agent_mcp(access_token: str):
         async with ClientSession(read, write) as session:
             await session.initialize()
             tools = await load_mcp_tools(session)
-            model = UiPathChatAnthropic(
-                model="anthropic.claude-3-5-sonnet-20240620-v1:0"
-            )
+            model = UiPathChat(model="anthropic.claude-3-5-sonnet-20240620-v1:0")
             agent = create_agent(model, tools=tools)
             yield agent
 

src/uipath_langchain/chat/__init__.py

@@ -18,9 +18,9 @@ def __getattr__(name):
 
         return UiPathBaseLLMClient
     if name == "UiPathChat":
-        from uipath_langchain_client.clients.openai import UiPathAzureChatOpenAI
+        from uipath_langchain_client.clients.normalized import UiPathNormalizedChatModel
 
-        return UiPathAzureChatOpenAI
+        return UiPathNormalizedChatModel
     if name == "UiPathAzureChatOpenAI":
         from uipath_langchain_client.clients.openai import UiPathAzureChatOpenAI
 

src/uipath_langchain/embeddings/__init__.py

@@ -13,6 +13,12 @@
 
 
 def __getattr__(name: str):
+    if name == "UiPathEmbeddings":
+        from uipath_langchain_client.clients.normalized import (
+            UiPathNormalizedEmbeddings,
+        )
+
+        return UiPathNormalizedEmbeddings
     if name == "UiPathAzureOpenAIEmbeddings":
         from uipath_langchain_client.clients.openai.embeddings import (
             UiPathAzureOpenAIEmbeddings,