Skip to content

Commit 9cde193

Browse files
authored
Update dependencies (#403)
1 parent 8b86463 commit 9cde193

File tree

3 files changed

+1006
-1162
lines changed

3 files changed

+1006
-1162
lines changed

backend/retrieval_graph/prompts.py

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,27 +1,36 @@
1-
from langchain import hub
1+
from langsmith import Client
22

33
"""Default prompts."""
44

5+
client = Client()
56
# fetch from langsmith
67
ROUTER_SYSTEM_PROMPT = (
7-
hub.pull("langchain-ai/chat-langchain-router-prompt").messages[0].prompt.template
8+
client.pull_prompt("langchain-ai/chat-langchain-router-prompt")
9+
.messages[0]
10+
.prompt.template
811
)
912
GENERATE_QUERIES_SYSTEM_PROMPT = (
10-
hub.pull("langchain-ai/chat-langchain-generate-queries-prompt")
13+
client.pull_prompt("langchain-ai/chat-langchain-generate-queries-prompt")
1114
.messages[0]
1215
.prompt.template
1316
)
1417
MORE_INFO_SYSTEM_PROMPT = (
15-
hub.pull("langchain-ai/chat-langchain-more-info-prompt").messages[0].prompt.template
18+
client.pull_prompt("langchain-ai/chat-langchain-more-info-prompt")
19+
.messages[0]
20+
.prompt.template
1621
)
1722
RESEARCH_PLAN_SYSTEM_PROMPT = (
18-
hub.pull("langchain-ai/chat-langchain-research-plan-prompt")
23+
client.pull_prompt("langchain-ai/chat-langchain-research-plan-prompt")
1924
.messages[0]
2025
.prompt.template
2126
)
2227
GENERAL_SYSTEM_PROMPT = (
23-
hub.pull("langchain-ai/chat-langchain-general-prompt").messages[0].prompt.template
28+
client.pull_prompt("langchain-ai/chat-langchain-general-prompt")
29+
.messages[0]
30+
.prompt.template
2431
)
2532
RESPONSE_SYSTEM_PROMPT = (
26-
hub.pull("langchain-ai/chat-langchain-response-prompt").messages[0].prompt.template
33+
client.pull_prompt("langchain-ai/chat-langchain-response-prompt")
34+
.messages[0]
35+
.prompt.template
2736
)

0 commit comments

Comments
 (0)