CS - first api hookups.

elizaOS · Nov 24, 2024 · b3d5718 · b3d5718
1 parent 666d5c1
commit b3d5718
Showing 1 changed file with 50 additions and 0 deletions.
diff --git a/packages/client-direct/src/index.ts b/packages/client-direct/src/index.ts
@@ -266,6 +266,56 @@ export class DirectClient {
                 res.json({ images: imagesRes });
             }
         );
+
+        this.app.post(
+            "/fine-tune",
+            async (req: express.Request, res: express.Response) => {
+                try {
+                    const response = await fetch('https://api.bageldb.ai/api/v1/asset', {
+                        method: 'POST',
+                        headers: {
+                            'Content-Type': 'application/json',
+                            'X-API-KEY': `${process.env.BAGEL_API_KEY}`
+                        },
+                        body: JSON.stringify(req.body)
+                    });
+
+                    const data = await response.json();
+                    res.json(data);
+                } catch (error) {
+                    res.status(500).json({ 
+                        error: 'Failed to forward request to BagelDB',
+                        details: error.message 
+                    });
+                }
+            }
+        );
+        this.app.get(
+            "/fine-tune/:assetId",
+            async (req: express.Request, res: express.Response) => {
+                const assetId = req.params.assetId;
+                try {
+                    const response = await fetch(`https://api.bageldb.ai/api/v1/asset/${assetId}/download`, {
+                        headers: {
+                            'X-API-KEY': `${process.env.BAGEL_API_KEY}`
+                        }
+                    });
+
+                    // Forward the content-type header
+                    res.set('Content-Type', response.headers.get('content-type'));
+
+                    // Convert ReadableStream to Buffer and send
+                    const arrayBuffer = await response.arrayBuffer();
+                    const buffer = Buffer.from(arrayBuffer);
+                    res.send(buffer);
+                } catch (error) {
+                    res.status(500).json({ 
+                        error: 'Failed to forward request to BagelDB',
+                        details: error.message 
+                    });
+                }
+            }
+        );
     }
 
     public registerAgent(runtime: AgentRuntime) {