import uvicorn
import requests
app = FastAPI()
# sparkai调用
from sparkai.llm.llm import ChatSparkLLM, ChunkPrintHandler
from sparkai.core.messages import ChatMessage
#星火认知大模型Spark Max的URL值，其他版本大模型URL值请前往文档（https://www.xfyun.cn/doc/spark/Web.html）查看
SPARKAI_URL = 'wss://spark-api.xf-yun.com/v1.1/chat'
#星火认知大模型调用秘钥信息，请前往讯飞开放平台控制台（https://console.xfyun.cn/services/bm35）查看
SPARKAI_APP_ID = '替换自己的ID'
SPARKAI_API_SECRET = '替换自己的SECRET'
SPARKAI_API_KEY = '替换自己的API_KEY'
#星火认知大模型Spark Max的domain值，其他版本大模型domain值请前往文档（https://www.xfyun.cn/doc/spark/Web.html）查看
SPARKAI_DOMAIN = 'general'
def search_sparkai(keyword):
spark = ChatSparkLLM(
spark_api_url=SPARKAI_URL,
spark_app_id=SPARKAI_APP_ID,
spark_api_key=SPARKAI_API_KEY,
spark_api_secret=SPARKAI_API_SECRET,
spark_llm_domain=SPARKAI_DOMAIN,
streaming=False,
)
messages = [ChatMessage(role="user",content=keyword)]
handler = ChunkPrintHandler()
generations = spark.generate([messages], callbacks=[handler])
data = str(generations)
return data
def get_text(t):
m=t.split("text='")[1]
if len(m) >1:
m=m.split("', message")[0]
else:
return t
return m
@app.get("/sparkai")
def sparkai_answer(keyword:str|None=None):
t=search_sparkai(keyword)
return get_text(t)
def zhongying(keywords):
zh = 'zh'
en = 'en'
for c in keywords:
if ('\u4e00'
