fix: mime-type was fixed to jpeg (#75)

callstackincubator · Dec 10, 2024 · e6ae07f · e6ae07f
1 parent a6d24d0
commit e6ae07f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/tools/src/vision.ts b/packages/tools/src/vision.ts
@@ -1,4 +1,5 @@
 import fs from 'node:fs/promises'
+import path from 'node:path'
 
 import { Provider } from '@dead-simple-ai-agent/framework/models'
 import { tool } from '@dead-simple-ai-agent/framework/tool'
@@ -8,7 +9,7 @@ import { z } from 'zod'
 
 const encodeImage = async (imagePath: string): Promise<string> => {
   const imageBuffer = await fs.readFile(imagePath)
-  return `data:image/jpeg;base64,${imageBuffer.toString('base64')}`
+  return `data:image/${path.extname(imagePath).toLowerCase().replace('.', '')};base64,${imageBuffer.toString('base64')}`
 }
 
 async function callOpenAI(