xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-12 00:36:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: AI对话基本完成

Browse Source
This commit is contained in:
RockChinQ
2024-01-27 21:50:40 +08:00
parent 850a4eeb7c
commit f10af09bd2
14 changed files with 308 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/openai/sysprompt/loaders/scenario.py
View File

@@ -21,14 +21,9 @@ class ScenarioPromptLoader(loader.PromptLoader):
file_json = json.loads(file_str)
messages = []
for msg in file_json["prompt"]:
role = llm_entities.MessageRole.SYSTEM
role = 'system'
if "role" in msg:
if msg["role"] == "user":
role = llm_entities.MessageRole.USER
elif msg["role"] == "system":
role = llm_entities.MessageRole.SYSTEM
elif msg["role"] == "function":
role = llm_entities.MessageRole.FUNCTION
role = msg['role']
messages.append(
llm_entities.Message(
role=role,

4
pkg/openai/sysprompt/loaders/single.py
View File

@@ -19,7 +19,7 @@ class SingleSystemPromptLoader(loader.PromptLoader):
name=name,
messages=[
llm_entities.Message(
role=llm_entities.MessageRole.SYSTEM,
role='system',
content=cnt
)
]
@@ -34,7 +34,7 @@ class SingleSystemPromptLoader(loader.PromptLoader):
name=file_name,
messages=[
llm_entities.Message(
role=llm_entities.MessageRole.SYSTEM,
role='system',
content=file_str
)
]