2023-06-30 17:10:48 +08:00
|
|
|
#!/usr/bin/env python
|
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
"""
|
|
|
|
|
@Time : 2023/5/11 14:45
|
|
|
|
|
@Author : alexanderwu
|
|
|
|
|
@File : llm.py
|
|
|
|
|
"""
|
|
|
|
|
|
2023-07-21 13:17:26 -04:00
|
|
|
from metagpt.provider.anthropic_api import Claude2 as Claude
|
2023-07-22 11:28:22 +08:00
|
|
|
from metagpt.provider.openai_api import OpenAIGPTAPI as LLM
|
2023-06-30 17:10:48 +08:00
|
|
|
|
|
|
|
|
DEFAULT_LLM = LLM()
|
2023-07-21 14:06:40 -04:00
|
|
|
CLAUDE_LLM = Claude()
|
2023-06-30 17:10:48 +08:00
|
|
|
|
2023-07-22 11:55:57 +08:00
|
|
|
|
2023-06-30 17:10:48 +08:00
|
|
|
async def ai_func(prompt):
|
|
|
|
|
"""使用LLM进行QA"""
|
|
|
|
|
return await DEFAULT_LLM.aask(prompt)
|
