refactor: 命令基本完成

This commit is contained in:
RockChinQ
2024-01-28 18:21:43 +08:00
parent 2a0cf57303
commit 1368ee22b2
21 changed files with 859 additions and 34 deletions

View File

@@ -102,7 +102,7 @@ class OpenAIChatCompletion(api.LLMAPIRequester):
m.dict(exclude_none=True) for m in conversation.prompt.messages
] + [m.dict(exclude_none=True) for m in conversation.messages]
req_messages.append({"role": "user", "content": str(query.message_chain)})
# req_messages.append({"role": "user", "content": str(query.message_chain)})
msg = await self._closure(req_messages, conversation)

View File

@@ -35,9 +35,16 @@ class PromptManager:
"""
return self.loader_inst.get_prompts()
async def get_prompt(self, name: str) -> loader.entities.Prompt:
async def get_prompt(self, name: str) -> loader.entities.Prompt:
"""获取Prompt
"""
for prompt in self.get_all_prompts():
if prompt.name == name:
return prompt
async def get_prompt_by_prefix(self, prefix: str) -> loader.entities.Prompt:
"""通过前缀获取Prompt
"""
for prompt in self.get_all_prompts():
if prompt.name.startswith(prefix):
return prompt