Merge pull request #70 from saagpatel/codex/refactor/use-llm-split

refactor(hooks): split useLlm into model / generation / streaming

Author: saagpatel

src/components/Draft/DraftTab.tsx

@@ -40,7 +40,8 @@ import { WorkspaceDialogs } from "./WorkspaceDialogs";
 import { WorkspaceModeShell } from "./WorkspaceModeShell";
 import { WorkspacePanels } from "./WorkspacePanels";
 import { WorkspaceWorkflowStrip } from "./WorkspaceWorkflowStrip";
-import { useLlm } from "../../hooks/useLlm";
+import { useLlmGeneration } from "../../hooks/useLlmGeneration";
+import { useLlmStreaming } from "../../hooks/useLlmStreaming";
 import { useDrafts } from "../../hooks/useDrafts";
 import { useKb } from "../../hooks/useKb";
 import { useAnalytics } from "../../hooks/useAnalytics";
@@ -101,11 +102,13 @@ export const DraftTab = forwardRef<DraftTabHandle, DraftTabProps>(
       isStreaming,
       clearStreamingText,
       cancelGeneration,
+    } = useLlmStreaming();
+    const {
       generateFirstResponse,
       generateChecklist,
       updateChecklist,
       generateWithContextParams,
-    } = useLlm();
+    } = useLlmGeneration();
     const {
       saveDraft,
       updateDraft,

src/components/Settings/SettingsTab.test.tsx

@@ -121,8 +121,8 @@ vi.mock("../shared/Button", () => ({
   ),
 }));
 
-vi.mock("../../hooks/useLlm", () => ({
-  useLlm: () => ({
+vi.mock("../../hooks/useLlmModel", () => ({
+  useLlmModel: () => ({
     loadModel: loadModelMock,
     unloadModel: unloadModelMock,
     getLoadedModel: getLoadedModelMock,

src/components/Settings/SettingsTab.tsx

@@ -14,7 +14,7 @@ import { useDownload } from "../../hooks/useDownload";
 import { useEmbedding } from "../../hooks/useEmbedding";
 import { useJira } from "../../hooks/useJira";
 import { useKb } from "../../hooks/useKb";
-import { useLlm } from "../../hooks/useLlm";
+import { useLlmModel } from "../../hooks/useLlmModel";
 import { useSearchApiEmbedding } from "../../hooks/useSearchApiEmbedding";
 import { useSettingsOps } from "../../hooks/useSettingsOps";
 import {
@@ -72,7 +72,7 @@ export function SettingsTab() {
     setContextWindow,
     loadCustomModel,
     validateGgufFile,
-  } = useLlm();
+  } = useLlmModel();
   const {
     setKbFolder,
     getKbFolder,