File size: 827 Bytes
151b478 10419ed 151b478 10419ed 151b478 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | from langchain_openai import ChatOpenAI
from dotenv import load_dotenv
from os import getenv
import os
# Initialize OpenAI model
from langchain_groq import ChatGroq
load_dotenv()
groq_api_key = os.environ['GROQ_API_KEY']
llm = ChatGroq(groq_api_key=groq_api_key, model_name="openai/gpt-oss-20b")
from typing import TypedDict, Annotated
from langgraph.graph import StateGraph, END, add_messages
class BasicChatState(TypedDict):
messages: Annotated[list, add_messages]
def chatbot(state: BasicChatState):
return {
"messages": [llm.invoke(state["messages"])]
}
# Create LangGraph nodes and edges
graph = StateGraph(BasicChatState)
graph.add_node("chatbot", chatbot)
graph.add_edge("chatbot", END)
# Set graph entry point
graph.set_entry_point("chatbot")
# Compile the graph
chat_agent = graph.compile() |