Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-13 12:55:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): unable to explain image when network search is active (#10228)

Browse Source 
Fix issue [PD-2316](https://linear.app/affine-design/issue/PD-2316).
This commit is contained in:
akumatus
2025-02-18 02:20:20 +00:00
parent eed00e0b26
commit 015452e8fb
3 changed files with 116 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
packages/frontend/core/src/blocksuite/presets/ai/chat-panel/chat-panel-input.ts
View File

@@ -288,7 +288,7 @@ export class ChatPanelInput extends SignalWatcher(WithDisposable(LitElement)) {
);
}
private get _promptName() {
private _getPromptName() {
if (this._isNetworkDisabled) {
return PROMPT_NAME_AFFINE_AI;
}
@@ -297,12 +297,12 @@ export class ChatPanelInput extends SignalWatcher(WithDisposable(LitElement)) {
: PROMPT_NAME_AFFINE_AI;
}
private async _updatePromptName() {
if (this._lastPromptName !== this._promptName) {
this._lastPromptName = this._promptName;
private async _updatePromptName(promptName: string) {
if (this._lastPromptName !== promptName) {
const sessionId = await this.getSessionId();
if (sessionId) {
await AIProvider.session?.updateSession(sessionId, this._promptName);
if (sessionId && AIProvider.session) {
await AIProvider.session.updateSession(sessionId, promptName);
this._lastPromptName = promptName;
}
}
}
@@ -457,7 +457,7 @@ export class ChatPanelInput extends SignalWatcher(WithDisposable(LitElement)) {
}}
@keydown=${async (evt: KeyboardEvent) => {
if (evt.key === 'Enter' && !evt.shiftKey && !evt.isComposing) {
this._onTextareaSend(evt);
await this._onTextareaSend(evt);
}
}}
@focus=${() => {
@@ -538,7 +538,7 @@ export class ChatPanelInput extends SignalWatcher(WithDisposable(LitElement)) {
</div>`;
}
private readonly _onTextareaSend = (e: MouseEvent | KeyboardEvent) => {
private readonly _onTextareaSend = async (e: MouseEvent | KeyboardEvent) => {
e.preventDefault();
e.stopPropagation();
@@ -549,17 +549,17 @@ export class ChatPanelInput extends SignalWatcher(WithDisposable(LitElement)) {
this.isInputEmpty = true;
this.textarea.style.height = 'unset';
this.send(value).catch(console.error);
await this.send(value);
};
send = async (text: string) => {
const { status, markdown, chips } = this.chatContextValue;
const { status, markdown, chips, images } = this.chatContextValue;
if (status === 'loading' || status === 'transmitting') return;
if (!text) return;
try {
const { images } = this.chatContextValue;
const { doc } = this.host;
const promptName = this._getPromptName();
this.updateContext({
images: [],
@@ -593,7 +593,9 @@ export class ChatPanelInput extends SignalWatcher(WithDisposable(LitElement)) {
],
});
await this._updatePromptName();
// must update prompt name after local chat message is updated
// otherwise, the unauthorized error can not be rendered properly
await this._updatePromptName(promptName);
const abortController = new AbortController();
const sessionId = await this.getSessionId();

117
packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-input.ts
View File

@@ -241,7 +241,7 @@ export class ChatBlockInput extends SignalWatcher(LitElement) {
${status === 'transmitting'
? html`<div @click=${this._handleAbort}>${ChatAbortIcon}</div>`
: html`<div
@click="${this._send}"
@click=${this._onTextareaSend}
class="chat-panel-send"
aria-disabled=${this._isInputEmpty}
>
@@ -306,7 +306,7 @@ export class ChatBlockInput extends SignalWatcher(LitElement) {
return !!this.chatContext.images.length;
}
private get _promptName() {
private _getPromptName() {
if (this._isNetworkDisabled) {
return PROMPT_NAME_AFFINE_AI;
}
@@ -315,15 +315,12 @@ export class ChatBlockInput extends SignalWatcher(LitElement) {
: PROMPT_NAME_AFFINE_AI;
}
private async _updatePromptName() {
if (this._lastPromptName !== this._promptName) {
this._lastPromptName = this._promptName;
private async _updatePromptName(promptName: string) {
if (this._lastPromptName !== promptName) {
const { currentSessionId } = this.chatContext;
if (currentSessionId) {
await AIProvider.session?.updateSession(
currentSessionId,
this._promptName
);
if (currentSessionId && AIProvider.session) {
await AIProvider.session.updateSession(currentSessionId, promptName);
this._lastPromptName = promptName;
}
}
}
@@ -346,7 +343,7 @@ export class ChatBlockInput extends SignalWatcher(LitElement) {
private readonly _handleKeyDown = async (evt: KeyboardEvent) => {
if (evt.key === 'Enter' && !evt.shiftKey && !evt.isComposing) {
evt.preventDefault();
await this._send();
await this._onTextareaSend(evt);
}
};
@@ -452,56 +449,70 @@ export class ChatBlockInput extends SignalWatcher(LitElement) {
`;
}
private readonly _send = async () => {
const { images, status } = this.chatContext;
if (status === 'loading' || status === 'transmitting') return;
private readonly _onTextareaSend = async (e: MouseEvent | KeyboardEvent) => {
e.preventDefault();
e.stopPropagation();
const text = this.textarea.value;
if (!text && !images.length) {
return;
}
const value = this.textarea.value.trim();
if (value.length === 0) return;
const { doc } = this.host;
this.textarea.value = '';
this._isInputEmpty = true;
this.textarea.style.height = 'unset';
this.updateContext({
images: [],
status: 'loading',
error: null,
});
const attachments = await Promise.all(
images?.map(image => readBlobAsURL(image))
);
await this._send(value);
};
const userInfo = await AIProvider.userInfo;
this.updateContext({
messages: [
...this.chatContext.messages,
{
id: '',
content: text,
role: 'user',
createdAt: new Date().toISOString(),
attachments,
userId: userInfo?.id,
userName: userInfo?.name,
avatarUrl: userInfo?.avatarUrl ?? undefined,
},
{
id: '',
content: '',
role: 'assistant',
createdAt: new Date().toISOString(),
},
],
});
const { currentChatBlockId, currentSessionId } = this.chatContext;
let content = '';
private readonly _send = async (text: string) => {
const { images, status, currentChatBlockId, currentSessionId } =
this.chatContext;
const chatBlockExists = !!currentChatBlockId;
let content = '';
if (status === 'loading' || status === 'transmitting') return;
if (!text) return;
try {
const { doc } = this.host;
const promptName = this._getPromptName();
this.updateContext({
images: [],
status: 'loading',
error: null,
});
const attachments = await Promise.all(
images?.map(image => readBlobAsURL(image))
);
const userInfo = await AIProvider.userInfo;
this.updateContext({
messages: [
...this.chatContext.messages,
{
id: '',
content: text,
role: 'user',
createdAt: new Date().toISOString(),
attachments,
userId: userInfo?.id,
userName: userInfo?.name,
avatarUrl: userInfo?.avatarUrl ?? undefined,
},
{
id: '',
content: '',
role: 'assistant',
createdAt: new Date().toISOString(),
},
],
});
// must update prompt name after local chat message is updated
// otherwise, the unauthorized error can not be rendered properly
await this._updatePromptName(promptName);
// If has not forked a chat session, fork a new one
let chatSessionId = currentSessionId;
if (!chatSessionId) {
@@ -518,8 +529,6 @@ export class ChatBlockInput extends SignalWatcher(LitElement) {
chatSessionId = forkSessionId;
}
await this._updatePromptName();
const abortController = new AbortController();
const stream = AIProvider.actions.chat?.({
input: text,