Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2024-12-22 16:48:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

simplify waveai imports

Browse Source
This commit is contained in:
Evan Simkowitz 2024-09-05 14:43:14 -07:00
parent 7393e5e7b6
commit 86685e4bcf
No known key found for this signature in database
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
frontend/app/view/waveai/waveai.tsx
View File

@ -4,14 +4,14 @@
import { Markdown } from "@/app/element/markdown"; import { Markdown } from "@/app/element/markdown";
import { TypingIndicator } from "@/app/element/typingindicator"; import { TypingIndicator } from "@/app/element/typingindicator";
import { atoms, fetchWaveFile, getUserName, globalStore, WOS } from "@/store/global"; import { atoms, fetchWaveFile, getUserName, globalStore, WOS } from "@/store/global";
import * as services from "@/store/services"; import { BlockService } from "@/store/services";
import { WshServer } from "@/store/wshserver"; import { WshServer } from "@/store/wshserver";
import { adaptFromReactOrNativeKeyEvent, checkKeyPressed } from "@/util/keyutil"; import { adaptFromReactOrNativeKeyEvent, checkKeyPressed } from "@/util/keyutil";
import * as util from "@/util/util"; import { isBlank } from "@/util/util";
import { atom, Atom, PrimitiveAtom, useAtomValue, useSetAtom, WritableAtom } from "jotai"; import { atom, Atom, PrimitiveAtom, useAtomValue, useSetAtom, WritableAtom } from "jotai";
import type { OverlayScrollbars } from "overlayscrollbars"; import type { OverlayScrollbars } from "overlayscrollbars";
import { OverlayScrollbarsComponent, OverlayScrollbarsComponentRef } from "overlayscrollbars-react"; import { OverlayScrollbarsComponent, OverlayScrollbarsComponentRef } from "overlayscrollbars-react";
import React, { forwardRef, useCallback, useEffect, useImperativeHandle, useMemo, useRef, useState } from "react"; import { forwardRef, memo, useCallback, useEffect, useImperativeHandle, useMemo, useRef, useState } from "react";
import tinycolor from "tinycolor2"; import tinycolor from "tinycolor2";
import "./waveai.less"; import "./waveai.less";
@ -107,9 +107,9 @@ export class WaveAiModel implements ViewModel {
}); });
this.viewText = atom((get) => { this.viewText = atom((get) => {
const settings = get(atoms.settingsAtom); const settings = get(atoms.settingsAtom);
const isCloud = util.isBlank(settings?.["ai:apitoken"]) && util.isBlank(settings?.["ai:baseurl"]); const isCloud = isBlank(settings?.["ai:apitoken"]) && isBlank(settings?.["ai:baseurl"]);
let modelText = "gpt-4o-mini"; let modelText = "gpt-4o-mini";
if (!isCloud && !util.isBlank(settings?.["ai:model"])) { if (!isCloud && !isBlank(settings?.["ai:model"])) {
modelText = settings["ai:model"]; modelText = settings["ai:model"];
} }
const viewTextChildren: HeaderElem[] = [ const viewTextChildren: HeaderElem[] = [
@ -198,11 +198,7 @@ export class WaveAiModel implements ViewModel {
role: "assistant", role: "assistant",
content: fullMsg, content: fullMsg,
}; };
const writeToHistory = services.BlockService.SaveWaveAiData(blockId, [ const writeToHistory = BlockService.SaveWaveAiData(blockId, [...history, newPrompt, responsePrompt]);
...history,
newPrompt,
responsePrompt,
]);
const typeResponse = simulateResponse(response); const typeResponse = simulateResponse(response);
Promise.all([writeToHistory, typeResponse]); Promise.all([writeToHistory, typeResponse]);
}; };
@ -279,7 +275,7 @@ interface ChatWindowProps {
messages: ChatMessageType[]; messages: ChatMessageType[];
} }
const ChatWindow = React.memo( const ChatWindow = memo(
forwardRef<OverlayScrollbarsComponentRef, ChatWindowProps>(({ chatWindowRef, messages }, ref) => { forwardRef<OverlayScrollbarsComponentRef, ChatWindowProps>(({ chatWindowRef, messages }, ref) => {
const [isUserScrolling, setIsUserScrolling] = useState(false); const [isUserScrolling, setIsUserScrolling] = useState(false);