diff options
Diffstat (limited to 'src/ollama.tsx')
| -rw-r--r-- | src/ollama.tsx | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/src/ollama.tsx b/src/ollama.tsx index c515574..945d2cb 100644 --- a/src/ollama.tsx +++ b/src/ollama.tsx @@ -152,16 +152,27 @@ export async function summarizeBlock() { export async function promptFromBlockEvent(b: IHookEvent) { try { const currentBlock = await logseq.Editor.getBlock(b.uuid) - const answerBlock = await logseq.Editor.insertBlock(currentBlock!.uuid, 'Generating....', { before: false }) - let response = ""; - const prompt = await promptLLM(`${currentBlock!.content}`); - await logseq.Editor.updateBlock(answerBlock!.uuid, `${prompt}`) + const answerBlock = await logseq.Editor.insertBlock(currentBlock!.uuid, '⌛Generating ...', { before: false }) + const response = await promptLLM(`${currentBlock!.content}`); + await logseq.Editor.updateBlock(answerBlock!.uuid, `${response}`) } catch (e: any) { logseq.UI.showMsg(e.toString(), 'warning') console.error(e) } } +export async function expandBlockEvent(b: IHookEvent) { + try { + const currentBlock = await logseq.Editor.getBlock(b.uuid) + const answerBlock = await logseq.Editor.insertBlock(currentBlock!.uuid, '⌛Generating ...', { before: false }) + const response = await promptLLM(`Expand: ${currentBlock!.content}`); + await logseq.Editor.updateBlock(answerBlock!.uuid, `${response}`) + } catch(e: any) { + logseq.UI.showMsg(e.toString(), 'warning') + console.error(e) + } +} + export async function askAI(prompt: string, context: string) { await delay(300) try { |