Merge pull request #8 from taweili/master

Add Prompt from Block Context Menu Command

1 files changed, 8 insertions, 1 deletions

diff --git a/src/App.tsx b/src/App.tsx
index f9592a3..b902629 100644
--- a/src/App.tsx
+++ b/src/App.tsx
@@ -1,6 +1,11 @@
 import React, { useEffect, useRef, useState } from "react";
 import { OllamaCommandPallete } from "./components/OllamaCommandPallete";
-import { convertToFlashCardFromEvent, DivideTaskIntoSubTasksFromEvent, ollamaUI, summarizeBlockFromEvent } from "./ollama";
+import {  convertToFlashCardFromEvent, 
+          DivideTaskIntoSubTasksFromEvent, 
+          ollamaUI, 
+          summarizeBlockFromEvent, 
+          promptFromBlockEvent,
+          expandBlockEvent } from "./ollama";
 import { useAppVisible } from "./utils";
 
 const options = [
@@ -43,6 +48,8 @@ function App() {
     logseq.Editor.registerBlockContextMenuItem("Create a flash card", convertToFlashCardFromEvent)
     logseq.Editor.registerBlockContextMenuItem("Summarize block", summarizeBlockFromEvent)
     logseq.Editor.registerBlockContextMenuItem("Divide into subtasks", DivideTaskIntoSubTasksFromEvent)
+    logseq.Editor.registerBlockContextMenuItem("Ollama: Prompt from Block", promptFromBlockEvent)
+    logseq.Editor.registerBlockContextMenuItem("Ollama: Expand Block", expandBlockEvent)
     logseq.App.registerCommandShortcut(
       { "binding": logseq.settings.shortcut },
       ollamaUI