diff --git a/dashboard/package.json b/dashboard/package.json index 3877cc5a46..c78c6c500c 100644 --- a/dashboard/package.json +++ b/dashboard/package.json @@ -15,6 +15,7 @@ "autoprefixer": "10.4.14", "eslint": "8.39.0", "eslint-config-next": "13.3.4", + "mongoose": "^7.1.0", "next": "13.3.4", "postcss": "8.4.23", "react": "18.2.0", diff --git a/dashboard/src/pages/api/fetch-llm-calls.ts b/dashboard/src/pages/api/fetch-llm-calls.ts new file mode 100644 index 0000000000..fc5e2633a5 --- /dev/null +++ b/dashboard/src/pages/api/fetch-llm-calls.ts @@ -0,0 +1,36 @@ +import mongoose from 'mongoose'; +import { NextApiRequest, NextApiResponse } from 'next'; +import connectMongo from './../../utils/connectMongo' + +// TODO: Fix issue to get all fields +// const LLMCallSchema = new mongoose.Schema({ +// prompt: { +// type: [[[String]]], +// required: true +// }, +// output: String, +// params: Object +// }); +const LLMCallSchema = new mongoose.Schema({}); +const LLMCallModel = mongoose.models.LLMCall || mongoose.model('LLMCall', LLMCallSchema, 'l_l_m_call'); + +export default async function handler( + req: NextApiRequest, + res: NextApiResponse +) { + + try { + console.log('CONNECTING TO MONGO'); + await connectMongo(); + console.log('CONNECTED TO MONGO'); + + const data = await LLMCallModel.find({}); + console.log(data) + return res.status(200).json(data); + } catch (error) { + console.log(error); + res.json({ message: 'error' }); + } + + +} \ No newline at end of file diff --git a/dashboard/src/utils/connectMongo.ts b/dashboard/src/utils/connectMongo.ts new file mode 100644 index 0000000000..ac4c1f72df --- /dev/null +++ b/dashboard/src/utils/connectMongo.ts @@ -0,0 +1,5 @@ +import mongoose from 'mongoose'; + +const connectMongo = async () => mongoose.connect(process.env.MONGO_URI || ''); + +export default connectMongo; diff --git a/docker-compose.yml b/docker-compose.yml index 47a751e9a8..4a0cd3e70b 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -53,7 +53,13 @@ services: ports: - "3000:3000" environment: - NODE_ENV: development + - NODE_ENV=development + - MONGO_URI=mongodb://username:password@mongo:27017/test?authSource=admin + networks: + - agenta-network + depends_on: + mongo: + condition: service_healthy networks: