add health-data-parser tag to vision parser

OpenHealthForAll · Feb 19, 2025 · a4da67b · a4da67b
1 parent bd1656c
commit a4da67b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/src/lib/health-data/parser/vision/google.ts b/src/lib/health-data/parser/vision/google.ts
@@ -45,7 +45,13 @@ export class GoogleVisionParser extends BaseVisionParser {
             date: HealthCheckupSchema.shape.date,
             name: HealthCheckupSchema.shape.name
         }));
-        const {date, name} = await messages.pipe(llm.withStructuredOutput(DateNameSchema, {method: 'functionCalling'}))
+        const {
+            date,
+            name
+        } = await messages.pipe(llm.withStructuredOutput(DateNameSchema, {method: 'functionCalling'}).withConfig({
+            runName: 'health-data-parser',
+            metadata: {input: options.input}
+        }))
             .withRetry({stopAfterAttempt: 3})
             .invoke(options.input)
 

diff --git a/src/lib/health-data/parser/vision/ollama.ts b/src/lib/health-data/parser/vision/ollama.ts
@@ -52,6 +52,9 @@ export class OllamaVisionParser extends BaseVisionParser {
         const messages = options.messages || ChatPromptTemplate.fromMessages([]);
         const chain = messages.pipe(llm.withStructuredOutput(HealthCheckupSchema, {
             method: 'functionCalling',
+        }).withConfig({
+            runName: 'health-data-parser',
+            metadata: {input: options.input}
         }))
         return await chain.withRetry({stopAfterAttempt: 3}).invoke(options.input);
     }

diff --git a/src/lib/health-data/parser/vision/openai.ts b/src/lib/health-data/parser/vision/openai.ts
@@ -35,6 +35,9 @@ export class OpenAIVisionParser extends BaseVisionParser {
         const messages = options.messages || ChatPromptTemplate.fromMessages([]);
         const chain = messages.pipe(llm.withStructuredOutput(HealthCheckupSchema, {
             method: 'functionCalling',
+        }).withConfig({
+            runName: 'health-data-parser',
+            metadata: {input: options.input}
         }))
         return await chain.withRetry({stopAfterAttempt: 3}).invoke(options.input);
     }