Skip to content

Commit 26359a1

Browse files
authored
Merge pull request #57 from auth0/today-system-prompt
chore(examples): ensure current date/time is loaded in system prompt
2 parents f8e2368 + d7f0713 commit 26359a1

1 file changed

Lines changed: 11 additions & 2 deletions

File tree

  • examples/calling-apis/langchain-examples/src/agents

examples/calling-apis/langchain-examples/src/agents/agent.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1+
from datetime import datetime
12
from typing import Annotated, Sequence, TypedDict
23

3-
from langchain_core.messages import AIMessage, BaseMessage
4+
from langchain_core.messages import AIMessage, BaseMessage, SystemMessage
45
from langchain_openai import ChatOpenAI
56
from langgraph.graph import END, START, StateGraph, add_messages
67
from langgraph.prebuilt import ToolNode
@@ -19,7 +20,15 @@ class State(TypedDict):
1920

2021

2122
async def call_llm(state: State):
22-
response = await llm.ainvoke(state["messages"])
23+
messages = state["messages"]
24+
system_prompt = f"""You are an assistant designed to answer random user's questions.
25+
**Additional Guidelines**:
26+
- Today’s date for reference: {datetime.now().isoformat()}
27+
"""
28+
system_message = SystemMessage(content=system_prompt)
29+
messages = [system_message] + messages
30+
31+
response = await llm.ainvoke(messages)
2332
return {"messages": [response]}
2433

2534

0 commit comments

Comments
 (0)