revert some imports

Author: cosminacho

samples/bring-your-own-model/main.py

@@ -1,7 +1,7 @@
 from langchain_core.messages import HumanMessage, SystemMessage
 from langgraph.graph import START, StateGraph, END
-from uipath_langchain_client.clients.openai import UiPathChatOpenAI
 from pydantic import BaseModel
+from uipath_langchain.chat import UiPathChat
 
 
 class GraphState(BaseModel):

samples/chat-uipath-agent/graph.py

@@ -1,7 +1,7 @@
 from langchain.agents import create_agent
 from langchain_community.tools import DuckDuckGoSearchResults
 
-from uipath_langchain_client.clients.openai import UiPathAzureChatOpenAI
+from uipath_langchain.chat import UiPathChatOpenAI
 search_tool = DuckDuckGoSearchResults()
 
 movie_system_prompt = """You are an advanced AI assistant specializing in movie research and analysis. Your primary functions are:
@@ -27,5 +27,5 @@
 DO NOT do any math calculations unless specifically related to movie statistics or box office figures.
 """
 
-llm = UiPathAzureChatOpenAI(model="gpt-4o-mini-2024-07-18")
+llm = UiPathChatOpenAI(model="gpt-4o-mini-2024-07-18")
 graph = create_agent(llm, tools=[search_tool], system_prompt=movie_system_prompt)

samples/email-organizer-agent/src/email_organizer/main.py

@@ -9,8 +9,7 @@
 from uipath.platform.common import CreateTask
 from email_organizer.outlook_client import OutlookClient
 from difflib import SequenceMatcher
-from uipath_langchain_client.clients.openai import UiPathAzureChatOpenAI
-
+from uipath_langchain.chat import UiPathChat
 # Configuration
 DEFAULT_CONFIDENCE = 0.0
 USER = 'me'
@@ -365,7 +364,7 @@ async def llm_node(state: GraphState) -> Command:
         """
 
 
-        llm = UiPathAzureChatOpenAI(model="gpt-4o-mini-2024-07-18")
+        llm = UiPathChat(model="gpt-4o-mini-2024-07-18")
 
         structured_llm = llm.with_structured_output(RuleSuggestions)
         response = await structured_llm.ainvoke(prompt)

samples/joke-agent/graph.py

@@ -9,7 +9,6 @@
 from uipath.core.guardrails import GuardrailScope
 
 from middleware import CustomFilterAction, LoggingMiddleware
-from uipath_langchain_client.clients.openai import UiPathAzureChatOpenAI
 from uipath_langchain.guardrails import (
     BlockAction,
     PIIDetectionEntity,
@@ -21,7 +20,7 @@
     UiPathPromptInjectionMiddleware,
 )
 from uipath_langchain.guardrails.actions import LoggingSeverityLevel
-
+from uipath_langchain.chat import UiPathChat
 
 # Define input schema for the agent
 class Input(BaseModel):
@@ -34,8 +33,8 @@ class Output(BaseModel):
     joke: str
 
 
-# Initialize UiPathAzureChatOpenAI LLM
-llm = UiPathAzureChatOpenAI(model="gpt-4o-2024-08-06", temperature=0.7)
+# Initialize UiPathChat LLM
+llm = UiPathChat(model="gpt-4o-2024-08-06", temperature=0.7)
 
 
 @tool