Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
L
LAE
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
文靖昊
LAE
Commits
afda82ab
Commit
afda82ab
authored
4 months ago
by
tinywell
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
补充当前时间
parent
5cb5d5df
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
2 deletions
+5
-2
agent_rate.py
src/server/agent_rate.py
+5
-2
No files found.
src/server/agent_rate.py
View file @
afda82ab
from
typing
import
Any
,
List
,
Sequence
,
Union
from
datetime
import
datetime
import
langchain_core
from
langchain.tools
import
BaseTool
from
langchain_core.prompts
import
PromptTemplate
,
ChatPromptTemplate
,
SystemMessagePromptTemplate
,
MessagesPlaceholder
,
HumanMessagePromptTemplate
...
...
@@ -123,15 +123,18 @@ Action:
你的回复格式为 Action:```$JSON_BLOB```然后 Observation。
"""
PROMPT_AGENT_HUMAN
=
"""{input}
\n\n
{agent_scratchpad}
\n
(请注意,无论如何都要以 JSON 对象回复。工具返回的数据必须使用表格展示,包含在最终输出中,并且要保证数据的完整性)"""
PROMPT_AGENT_SYS_VARS
=
[
"tool_names"
,
"tools"
]
class
RateAgentV2
:
def
__init__
(
self
,
llm
,
tools
:
List
[
BaseTool
],
prompt
:
PromptTemplate
=
None
,
verbose
:
bool
=
False
,
**
args
):
date_now
=
datetime
.
now
()
.
strftime
(
"
%
Y-
%
m-
%
d"
)
prompt_human
=
f
"{PROMPT_AGENT_HUMAN}
\n\n
今天是{date_now}"
prompt
=
ChatPromptTemplate
.
from_messages
([
SystemMessagePromptTemplate
.
from_template
(
ONLINE_RATE_SYSTEM_PROMPT
),
MessagesPlaceholder
(
variable_name
=
"chat_history"
,
optional
=
True
),
HumanMessagePromptTemplate
.
from_template
(
PROMPT_AGENT_HUMAN
)
HumanMessagePromptTemplate
.
from_template
(
prompt_human
)
])
agent
=
create_structured_chat_agent
(
llm
,
tools
,
prompt
)
...
...
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment