GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-20 05:20:15 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix stream output

Browse Source
This commit is contained in:
Zhang Minghan 2023-10-24 20:28:29 +08:00
parent 2044e1bbe6
commit dc96278fda
4 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
adapter/chatgpt/chat.go
View File

@ -57,7 +57,7 @@ func (c *ChatInstance) ProcessLine(data string) string {
var form *ChatStreamResponse var form *ChatStreamResponse
if form = utils.UnmarshalForm[ChatStreamResponse](item); form == nil { if form = utils.UnmarshalForm[ChatStreamResponse](item); form == nil {
if form = utils.UnmarshalForm[ChatStreamResponse](item); form == nil { if form = utils.UnmarshalForm[ChatStreamResponse](item[:len(item)-1]); form == nil {
fmt.Println(fmt.Sprintf("chatgpt error: cannot parse response: %s", item)) fmt.Println(fmt.Sprintf("chatgpt error: cannot parse response: %s", item))
return "" return ""
} }

7
adapter/claude/chat.go
View File

@ -77,15 +77,18 @@ func (c *ChatInstance) CreateStreamChatRequest(props *ChatProps, hook globals.Ho
// response example: // response example:
// //
// event:completion // event:completion
// {"completion":"么","stop_reason":null,"model":"claude-1.3","stop":null,"log_id":"605b37b2bbab7a557b811ce47d8beeab959cbab22e46b515cb15e3f00afcbe24"}
// data:{"completion":"!","stop_reason":null,"model":"claude-2.0","stop":null,"log_id":"f5f659a5807419c94cfac4a9f2f79a66e95733975714ce7f00e30689dd136b02"} // data:{"completion":"!","stop_reason":null,"model":"claude-2.0","stop":null,"log_id":"f5f659a5807419c94cfac4a9f2f79a66e95733975714ce7f00e30689dd136b02"}
if !strings.HasPrefix(data, "data:") { if !strings.HasPrefix(data, "data:") && strings.HasPrefix(data, "event:") {
return nil return nil
} else { } else {
data = strings.TrimSpace(strings.TrimPrefix(data, "data:")) data = strings.TrimSpace(strings.TrimPrefix(data, "data:"))
} }
if len(data) == 0 {
return nil
}
if form := utils.UnmarshalForm[ChatResponse](data); form != nil { if form := utils.UnmarshalForm[ChatResponse](data); form != nil {
if err := hook(form.Completion); err != nil { if err := hook(form.Completion); err != nil {
return err return err

2
app/src/conf.ts
View File

@ -1,7 +1,7 @@
import axios from "axios"; import axios from "axios";
import { Model } from "./conversation/types.ts"; import { Model } from "./conversation/types.ts";
export const version = "3.5.4"; export const version = "3.5.5";
export const dev: boolean = window.location.hostname === "localhost"; export const dev: boolean = window.location.hostname === "localhost";
export const deploy: boolean = true; export const deploy: boolean = true;
export let rest_api: string = "http://localhost:8094"; export let rest_api: string = "http://localhost:8094";

4
app/src/store/quota.ts
View File

@ -52,7 +52,7 @@ export const quotaSelector = (state: RootState): string =>
const refreshQuota = async (dispatch: AppDispatch) => { const refreshQuota = async (dispatch: AppDispatch) => {
const current = new Date().getTime(); //@ts-ignore const current = new Date().getTime(); //@ts-ignore
if (window.hasOwnProperty("quota") && current - window.quota < 2500) return; //@ts-ignore if (window.hasOwnProperty("quota") && current - window.quota < 5000) return; //@ts-ignore
window.quota = current; window.quota = current;
const response = await axios.get("/quota"); const response = await axios.get("/quota");
@ -60,6 +60,6 @@ const refreshQuota = async (dispatch: AppDispatch) => {
}; };
export const refreshQuotaTask = (dispatch: AppDispatch) => { export const refreshQuotaTask = (dispatch: AppDispatch) => {
setInterval(() => refreshQuota(dispatch), 5000); setInterval(() => refreshQuota(dispatch), 8000);
refreshQuota(dispatch).then(); refreshQuota(dispatch).then();
}; };