Moar TS fixes

jozu-ai · Dec 3, 2024 · 0db8f7f · 0db8f7f
1 parent fefa761
commit 0db8f7f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/frontend/dev-mode/src/components/ChatPrompt.vue b/frontend/dev-mode/src/components/ChatPrompt.vue
@@ -15,7 +15,8 @@ import {
   type Parameters,
   DEFAULT_SESSION,
   DEFAULT_PARAMS_VALUES,
-  type TranscriptMessage
+  type TranscriptMessage,
+  type LlamaComposableResponse
 } from '@/composables/useLlama'
 
 const message = ref('')
@@ -31,10 +32,10 @@ const stats = inject('stats', {} as Stats)
 const isGenerating = inject('isGenerating', false)
 const isPending = inject('isPending', false)
 const isChatStarted = inject('isChatStarted', false)
-const runChat = inject<Function>('runChat')
-const runCompletion = inject<Function>('runCompletion')
-const stop = inject<Function>('stop')
-const uploadImage = inject<Function>('uploadImage')
+const runChat = inject<LlamaComposableResponse['chat']>('runChat')
+const runCompletion = inject<LlamaComposableResponse['runCompletion']>('runCompletion')
+const stop = inject<LlamaComposableResponse['stop']>('stop')
+const uploadImage = inject<LlamaComposableResponse['uploadImage']>('uploadImage')
 
 if (!runChat || !runCompletion || !stop || !uploadImage) {
   throw new Error('required function not provided')

diff --git a/frontend/dev-mode/src/composables/useLlama.ts b/frontend/dev-mode/src/composables/useLlama.ts
@@ -59,7 +59,7 @@ export type Parameters = {
   prop_order?: string,
 }
 
-type LlamaComposableResponse = {
+export type LlamaComposableResponse = {
   stats: Ref<Record<string, string> | null>,
   session: Ref<Session>,
   template: (str: string, extraSettings?: Record<string, any>) => string,