Spaces:
Sleeping
Sleeping
Update services/graph_service.py
Browse files- 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 |
-
|
| 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.
|
| 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 |
-
|
| 70 |
prompt = ChatPromptTemplate.from_messages([
|
| 71 |
-
|
| 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 |
-
|
|
|
|
| 78 |
|
| 79 |
def _agent_node(self, state: AgentState):
|
| 80 |
-
"""Prépare le prompt et appelle 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,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,
|