QuentinL52 commited on
Commit
a1a6bdc
·
verified ·
1 Parent(s): d426b96

Update services/graph_service.py

Browse files
Files changed (1) hide show
  1. services/graph_service.py +15 -14
services/graph_service.py CHANGED
@@ -6,7 +6,7 @@ from typing import TypedDict, Annotated, Sequence, Dict, Any, List
6
  from langchain_openai import ChatOpenAI
7
  from langchain_core.runnables import Runnable
8
  from langchain_core.messages import BaseMessage, AIMessage, HumanMessage, SystemMessage
9
- from langchain.agents import create_openai_tools_agent
10
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
11
  from langgraph.graph import StateGraph, END
12
  from langgraph.prebuilt import ToolNode
@@ -17,7 +17,7 @@ class AgentState(TypedDict):
17
  messages: Annotated[Sequence[BaseMessage], lambda x, y: x + y]
18
  user_id: str
19
  job_offer_id: str
20
-
21
  class GraphInterviewProcessor:
22
  """
23
  Cette classe encapsule la logique d'un entretien en utilisant LangGraph.
@@ -39,7 +39,7 @@ class GraphInterviewProcessor:
39
  self.skills_summary = self._extract_skills_summary()
40
  self.reconversion_info = self._extract_reconversion_info()
41
 
42
- self.agent = self._create_agent()
43
  self.graph = self._build_graph()
44
  logging.info("GraphInterviewProcessor initialisé avec succès.")
45
 
@@ -65,19 +65,19 @@ class GraphInterviewProcessor:
65
  if reconversion.get('is_reconversion'):
66
  return f"CANDIDAT EN RECONVERSION: {reconversion.get('analysis', '')}"
67
  return "Le candidat n'est pas identifié comme étant en reconversion."
68
-
69
- def _create_agent(self) -> Runnable:
70
  prompt = ChatPromptTemplate.from_messages([
71
- SystemMessage(content="{system_prompt_content}"),
72
  MessagesPlaceholder(variable_name="messages"),
73
- MessagesPlaceholder(variable_name="agent_scratchpad"),
74
  ])
75
  llm = ChatOpenAI(api_key=os.getenv("OPENAI_API_KEY"), model="gpt-4o-mini", temperature=0.7)
76
  tools = [trigger_interview_analysis]
77
- return create_openai_tools_agent(llm, tools, prompt)
 
78
 
79
  def _agent_node(self, state: AgentState):
80
- """Prépare le prompt et appelle l'agent pour obtenir une décision (parler ou utiliser un outil)."""
81
  system_prompt_content = self.system_prompt_template.format(
82
  entreprise=self.job_offer.get('entreprise', 'notre entreprise'),
83
  poste=self.job_offer.get('poste', 'ce poste'),
@@ -89,12 +89,9 @@ class GraphInterviewProcessor:
89
  skills_analysis=self.skills_summary,
90
  reconversion_analysis=self.reconversion_info
91
  )
92
-
93
- response = self.agent.invoke({
94
  "system_prompt_content": system_prompt_content,
95
  "messages": state["messages"],
96
- "user_id": state["user_id"],
97
- "job_offer_id": state["job_offer_id"],
98
  })
99
  return {"messages": [response]}
100
 
@@ -124,7 +121,7 @@ class GraphInterviewProcessor:
124
  "end_turn": END
125
  }
126
  )
127
-
128
  graph.add_edge("tools", "agent")
129
 
130
  return graph.compile()
@@ -133,6 +130,10 @@ class GraphInterviewProcessor:
133
  """Point d'entrée pour lancer une conversation dans le graphe."""
134
  langchain_messages = [HumanMessage(content=m["content"]) if m["role"] == "user" else AIMessage(content=m["content"]) for m in messages]
135
 
 
 
 
 
136
  initial_state = {
137
  "user_id": self.user_id,
138
  "job_offer_id": self.job_offer_id,
 
6
  from langchain_openai import ChatOpenAI
7
  from langchain_core.runnables import Runnable
8
  from langchain_core.messages import BaseMessage, AIMessage, HumanMessage, SystemMessage
9
+ # create_openai_tools_agent n'est plus nécessaire
10
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
11
  from langgraph.graph import StateGraph, END
12
  from langgraph.prebuilt import ToolNode
 
17
  messages: Annotated[Sequence[BaseMessage], lambda x, y: x + y]
18
  user_id: str
19
  job_offer_id: str
20
+
21
  class GraphInterviewProcessor:
22
  """
23
  Cette classe encapsule la logique d'un entretien en utilisant LangGraph.
 
39
  self.skills_summary = self._extract_skills_summary()
40
  self.reconversion_info = self._extract_reconversion_info()
41
 
42
+ self.agent_runnable = self._create_agent_runnable()
43
  self.graph = self._build_graph()
44
  logging.info("GraphInterviewProcessor initialisé avec succès.")
45
 
 
65
  if reconversion.get('is_reconversion'):
66
  return f"CANDIDAT EN RECONVERSION: {reconversion.get('analysis', '')}"
67
  return "Le candidat n'est pas identifié comme étant en reconversion."
68
+ def _create_agent_runnable(self) -> Runnable:
69
+ """Crée une chaîne (runnable) qui agit comme notre agent."""
70
  prompt = ChatPromptTemplate.from_messages([
71
+ ("system", "{system_prompt_content}"),
72
  MessagesPlaceholder(variable_name="messages"),
 
73
  ])
74
  llm = ChatOpenAI(api_key=os.getenv("OPENAI_API_KEY"), model="gpt-4o-mini", temperature=0.7)
75
  tools = [trigger_interview_analysis]
76
+ llm_with_tools = llm.bind_tools(tools)
77
+ return prompt | llm_with_tools
78
 
79
  def _agent_node(self, state: AgentState):
80
+ """Prépare le prompt et appelle le runnable de l'agent."""
81
  system_prompt_content = self.system_prompt_template.format(
82
  entreprise=self.job_offer.get('entreprise', 'notre entreprise'),
83
  poste=self.job_offer.get('poste', 'ce poste'),
 
89
  skills_analysis=self.skills_summary,
90
  reconversion_analysis=self.reconversion_info
91
  )
92
+ response = self.agent_runnable.invoke({
 
93
  "system_prompt_content": system_prompt_content,
94
  "messages": state["messages"],
 
 
95
  })
96
  return {"messages": [response]}
97
 
 
121
  "end_turn": END
122
  }
123
  )
124
+
125
  graph.add_edge("tools", "agent")
126
 
127
  return graph.compile()
 
130
  """Point d'entrée pour lancer une conversation dans le graphe."""
131
  langchain_messages = [HumanMessage(content=m["content"]) if m["role"] == "user" else AIMessage(content=m["content"]) for m in messages]
132
 
133
+ if not langchain_messages:
134
+ logging.info("Historique de conversation vide. Ajout d'un message de démarrage interne.")
135
+ langchain_messages.append(HumanMessage(content="Bonjour, je suis prêt à commencer l'entretien."))
136
+
137
  initial_state = {
138
  "user_id": self.user_id,
139
  "job_offer_id": self.job_offer_id,