chore: cleanup

rpidanny · Jul 3, 2024 · ef42500 · ef42500
1 parent a5c866f
commit ef42500
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 7 deletions.
diff --git a/jest.config.ts b/jest.config.ts
@@ -48,7 +48,7 @@ const config: JestConfigWithTsJest = {
     global: {
       statements: 81,
       branches: 90,
-      functions: 92,
+      functions: 90,
       lines: 81,
     },
   },

diff --git a/src/commands/chat/index.ts b/src/commands/chat/index.ts
@@ -69,9 +69,7 @@ export default class Chat extends BaseCommand<typeof Chat> {
 
     initChatContainer(
       {
-        headless: false,
         concurrency,
-        summarize: false,
         llmProvider,
         skipCaptcha,
         legacyProcessing,

diff --git a/src/commands/config/set.ts b/src/commands/config/set.ts
@@ -23,7 +23,7 @@ export default class SetConfig extends Command {
     const openai = await uiInput.promptOpenAIConfig(existingConfig?.openai)
     const ollama = await uiInput.promptOllamaConfig(existingConfig?.ollama)
 
-    const config = {
+    const config: TConfig = {
       openai,
       ollama,
     }

diff --git a/src/commands/download/papers.ts b/src/commands/download/papers.ts
@@ -52,6 +52,7 @@ export default class DownloadPapers extends BaseCommand<typeof DownloadPapers> {
     const { headless } = this.flags
 
     initDownloadContainer({ headless }, this.logger)
+
     this.odysseus = Container.get(Odysseus)
     await this.odysseus.init()
 

diff --git a/src/containers/chat.container.ts b/src/containers/chat.container.ts
@@ -8,17 +8,23 @@ import { initSearchContainer } from './search.container.js'
 
 export function initChatContainer(
   opts: {
-    headless: boolean
     concurrency: number
-    summarize: boolean
     llmProvider: LLMProvider
     skipCaptcha: boolean
     legacyProcessing: boolean
   },
   config: TConfig,
   logger: Quill,
 ) {
-  initSearchContainer(opts, config, logger)
+  initSearchContainer(
+    {
+      ...opts,
+      headless: false,
+      summarize: false,
+    },
+    config,
+    logger,
+  )
 
   Container.set(ChatOpenAI, Container.get(LLMFactory).getLLM(LLMProvider.OpenAI, config))
 }