Merge pull request #28 from Avivbens/feat/support-source-language-detection-in-translate

Author: Avivbens

projects/packages/text-transformer/info.plist

@@ -2434,7 +2434,7 @@ https://github.com/Avivbens/alfredo</string>
       </dict>
     </array>
     <key>version</key>
-    <string>5.4.1</string>
+    <string>5.4.2</string>
     <key>webaddress</key>
     <string>https://github.com/Avivbens/alfredo</string>
   </dict>

projects/packages/text-transformer/src/main/translate/translate.ts

@@ -2,11 +2,12 @@ import type { AlfredListItem } from 'fast-alfred';
 import { FastAlfred } from 'fast-alfred';
 import { setTimeout } from 'node:timers/promises';
 import { getActiveApp } from '@alfredo/active-app';
-import { AvailableModels, callModel } from '@alfredo/llm';
+import { AvailableModels, callModelWithStructuredResponse } from '@alfredo/llm';
 import { registerUpdater } from '@alfredo/updater';
 import { DEFAULT_DEBOUNCE_TIME, LANGUAGE_DELIMITER } from '../../common/defaults.constants';
 import { TRANSLATE_SYSTEM_PROMPT } from '../../common/prompts/translate.prompt';
 import { Variables } from '../../common/variables.enum';
+import { translateResultSchema } from '../../models/translate-results.schema';
 
 interface ParsedTranslationInput {
   targetLanguage: string;
@@ -88,13 +89,22 @@ function parseLanguageFromInput(input: string): ParsedTranslationInput {
       targetLanguage,
     });
 
-    const res = await callModel(token, model, { system, user: textToTranslate });
-
+    /**
+     * Call the model with structured response parsing
+     */
+    const { translation, sourceLanguage } = await callModelWithStructuredResponse(
+      token,
+      model,
+      { system, user: textToTranslate },
+      translateResultSchema,
+    );
+
+    const subtitle = `${sourceLanguage} => ${targetLanguage}`;
     const items: AlfredListItem[] = [
       {
-        title: res,
-        subtitle: 'Translate',
-        arg: res,
+        title: translation,
+        subtitle,
+        arg: translation,
       },
     ];
 

projects/packages/text-transformer/src/models/translate-results.schema.ts

@@ -0,0 +1,10 @@
+import z from 'zod';
+
+export const translateResultSchema = z.object({
+  translation: z.string(),
+  sourceLanguage: z
+    .string()
+    .describe(
+      'The original language of the provided text. Should be the full name of the language, e.g., "English", "Spanish", "French".',
+    ),
+});