code.py

## agent 2

## chatbot with some memory

## objectives :
# 1. use different message types - HumanMessage and ai message
# 2. maintain full conversation history using both message types
# 3. use gpt-4o model using langchain chatopenai , or langchain-google-genai


# main goal : create a form of memory for our agent

import os
from typing import TypedDict, List , Union
from langchain_core.messages import HumanMessage, AIMessage

from langchain_google_genai import ChatGoogleGenerativeAI
from langgraph.graph import StateGraph , START, END
from dotenv import load_dotenv

load_dotenv()

class AgentState(TypedDict):
  messages : List[Union[HumanMessage,AIMessage]]


llm = ChatGoogleGenerativeAI(
     model="gemini-pro",
    google_api_key=os.getenv("GEMINI_API_KEY")
)



def process(state:AgentState) -> AgentState :
  """this node will solve the request you input"""
  response = llm.invoke(state['messages'])

  state["messages"].append(AIMessage(content=response.content)) # only extract contents
  # this is our memory

  print(f"\nAI : {response.content}")
  print("CUURENT STATE: ", state["messages"])
  return state


graph = StateGraph(AgentState)
graph.add_node("process",process)

graph.add_edge(START,"process")
graph.add_edge('process',END)

agent = graph.compile()


conversation_history = []
user_input = input("Enter : ")


while user_input !=  "exit" :
  conversation_history.append(HumanMessage(content=user_input))
  result = agent.invoke({"messages": conversation_history})

 # print(result["messages"])
  conversation_history = result["messages"]

  user_input = input("Enter: ")



with open("logging.txt", "w") as file :
  file.write("Your conversation Log:\n")

  for message in conversation_history:
    if isinstance(message, HumanMessage):
      file.write(f"You : {message.content}\n")
    elif isinstance(message, AIMessage):
      file.write(f"AI: {message.content}\n\n")

  file.write("End of Conversation")

print("Conversation saved to logging.txt")



### here we have 2 big problems here

# solution for you data being get wiped out ?
# is , you can store it in DB , textfiles , vector db