mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-08-14 14:31:00 +00:00
feat: add shengsuanyun requester
This commit is contained in:
@@ -0,0 +1,32 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import openai
|
||||||
|
import typing
|
||||||
|
|
||||||
|
from . import chatcmpl
|
||||||
|
import openai.types.chat.chat_completion as chat_completion
|
||||||
|
|
||||||
|
|
||||||
|
class ShengSuanYunChatCompletions(chatcmpl.OpenAIChatCompletions):
|
||||||
|
"""胜算云 ChatCompletion API 请求器"""
|
||||||
|
|
||||||
|
client: openai.AsyncClient
|
||||||
|
|
||||||
|
default_config: dict[str, typing.Any] = {
|
||||||
|
'base_url': 'https://router.shengsuanyun.com/api/v1',
|
||||||
|
'timeout': 120,
|
||||||
|
}
|
||||||
|
|
||||||
|
async def _req(
|
||||||
|
self,
|
||||||
|
args: dict,
|
||||||
|
extra_body: dict = {},
|
||||||
|
) -> chat_completion.ChatCompletion:
|
||||||
|
return await self.client.chat.completions.create(
|
||||||
|
**args,
|
||||||
|
extra_body=extra_body,
|
||||||
|
extra_headers={
|
||||||
|
'HTTP-Referer': 'https://langbot.app',
|
||||||
|
'X-Title': 'LangBot',
|
||||||
|
},
|
||||||
|
)
|
||||||
File diff suppressed because one or more lines are too long
|
After Width: | Height: | Size: 7.4 KiB |
@@ -0,0 +1,38 @@
|
|||||||
|
apiVersion: v1
|
||||||
|
kind: LLMAPIRequester
|
||||||
|
metadata:
|
||||||
|
name: shengsuanyun-chat-completions
|
||||||
|
label:
|
||||||
|
en_US: ShengSuanYun
|
||||||
|
zh_Hans: 胜算云
|
||||||
|
icon: shengsuanyun.svg
|
||||||
|
spec:
|
||||||
|
config:
|
||||||
|
- name: base_url
|
||||||
|
label:
|
||||||
|
en_US: Base URL
|
||||||
|
zh_Hans: 基础 URL
|
||||||
|
type: string
|
||||||
|
required: true
|
||||||
|
default: "https://router.shengsuanyun.com/api/v1"
|
||||||
|
- name: args
|
||||||
|
label:
|
||||||
|
en_US: Args
|
||||||
|
zh_Hans: 附加参数
|
||||||
|
type: object
|
||||||
|
required: true
|
||||||
|
default: {}
|
||||||
|
- name: timeout
|
||||||
|
label:
|
||||||
|
en_US: Timeout
|
||||||
|
zh_Hans: 超时时间
|
||||||
|
type: int
|
||||||
|
required: true
|
||||||
|
default: 120
|
||||||
|
support_type:
|
||||||
|
- llm
|
||||||
|
- text-embedding
|
||||||
|
execution:
|
||||||
|
python:
|
||||||
|
path: ./shengsuanyun.py
|
||||||
|
attr: ShengSuanYunChatCompletions
|
||||||
Reference in New Issue
Block a user