From b3d57180274fe3a41d67529fc994f39c8b37cff4 Mon Sep 17 00:00:00 2001
From: justabot <colin@bagel.net>
Date: Sun, 24 Nov 2024 12:11:24 -0500
Subject: [PATCH] CS - first api hookups.
---
packages/client-direct/src/index.ts | 50 +++++++++++++++++++++++++++++
1 file changed, 50 insertions(+)
diff --git a/packages/client-direct/src/index.ts b/packages/client-direct/src/index.ts
index 7ddc5fa233f..ec6eb1badd8 100644
--- 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) {