chore: Refactor AI service to fix typo in downloaded variable name an…

…d improve model loading status display
coasys · Oct 4, 2024 · a3ccd45 · a3ccd45
1 parent 4e24ea8
commit a3ccd45
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/core/src/ai/AIClient.ts b/core/src/ai/AIClient.ts
@@ -193,20 +193,15 @@ export class AIClient {
             }
         }));
 
-        const subscription = await this.#apolloClient.subscribe({
-            query: gql`
-                subscription AiTranscriptionText($streamId: String!) {
-                    aiTranscriptionText(streamId: $streamId)
-                }
-            `,
-            variables: {
-                streamId: aiOpenTranscriptionStream
-            }
+        const subscription = this.#apolloClient.subscribe({
+            query: gql` subscription {
+                aiTranscriptionText(streamId: "${aiOpenTranscriptionStream}")
+            }`
         }).subscribe({
             next(data) {
-                streamCallback(data.data.transcriptionText);
+                streamCallback(data.data.aiTranscriptionText);
 
-                return data.data.transcriptionText;
+                return data.data.aiTranscriptionText;
             },
             error(err) {
                 console.error(err);