WIP

app/main.wasp

@@ -265,6 +265,10 @@ action updateUserById {
   entities: [User]
 }
 
+action getModels {
+  fn: import { getModels } from "@src/server/actions.js",
+  entities: []
+}
 
 // 📚 Queries
 

app/src/client/App.tsx

@@ -59,7 +59,6 @@ export default function App({ children }: { children: ReactNode }) {
 
   useEffect(() => {
     if (user) {
-      console.log('user', user);
       if (!user.isSignUpComplete) {
         if (user.hasAcceptedTos) {
           updateCurrentUser({

app/src/client/app/ModelsPage.tsx

@@ -1,7 +1,43 @@
+import { useState } from 'react';
 import CustomLayout from './layout/CustomLayout';
 import CustomBreadcrumb from '../components/CustomBreadcrumb';
+import Button from '../components/Button';
+import { getModels } from 'wasp/client/operations';
+
+interface JsonSchemaProperty {
+  default?: string;
+  description: string;
+  enum?: string[];
+  title: string;
+  type: string;
+  const?: string;
+  format?: string;
+  maxLength?: number;
+  minLength?: number;
+}
+
+interface JsonSchema {
+  properties: Record<string, JsonSchemaProperty>;
+  required: string[];
+  title: string;
+  type: string;
+}
+
+interface Schema {
+  name: string;
+  json_schema: JsonSchema;
+}
+
+interface ModelSchema {
+  schemas: Schema[];
+}
 
 const ModelsPage = () => {
+  const [modelsSchema, setModelsSchema] = useState<ModelSchema | null>(null);
+  const logout = async () => {
+    const response = await getModels();
+    setModelsSchema(response);
+  };
   return (
     <CustomLayout>
       <CustomBreadcrumb pageName='Models' />
@@ -12,7 +48,8 @@ const ModelsPage = () => {
               className='flex-col flex items-start justify-between p-6 gap-3 w-full'
               style={{ width: '1000px', height: '600px' }}
             >
-              <span className='text-sm font-medium text-airt-primary'>Some content goes here...</span>
+              <Button onClick={logout} label='Add Model' />
+              <p>{modelsSchema?.schemas.map((schema: any) => schema.name)}</p>
             </div>
           </div>
         </div>

app/src/server/actions.ts

@@ -1,11 +1,18 @@
 import { type User } from 'wasp/entities';
 import { HttpError } from 'wasp/server';
-import { type StripePayment, type UpdateCurrentUser, type UpdateUserById } from 'wasp/server/operations';
+import {
+  type StripePayment,
+  type UpdateCurrentUser,
+  type UpdateUserById,
+  type GetModels,
+} from 'wasp/server/operations';
 import Stripe from 'stripe';
 import type { StripePaymentResult } from '../shared/types';
 import { fetchStripeCustomer, createStripeCheckoutSession } from './payments/stripeUtils.js';
 import { TierIds } from '../shared/constants.js';
 
+const FASTAGENCY_SERVER_URL = process.env.FASTAGENCY_SERVER_URL || 'http://127.0.0.1:8000';
+
 export const stripePayment: StripePayment<string, StripePaymentResult> = async (tier, context) => {
   if (!context.user || !context.user.email) {
     throw new HttpError(401);
@@ -82,3 +89,27 @@ export const updateCurrentUser: UpdateCurrentUser<Partial<User>, User> = async (
     data: user,
   });
 };
+
+export const getModels: GetModels<void, any> = async (user, context) => {
+  if (!context.user) {
+    throw new HttpError(401);
+  }
+
+  try {
+    const response = await fetch(`${FASTAGENCY_SERVER_URL}/models/llms/schemas`, {
+      method: 'GET',
+      headers: { 'Content-Type': 'application/json' },
+    });
+    const json: any = (await response.json()) as { detail?: string }; // Parse JSON once
+
+    if (!response.ok) {
+      const errorMsg = json.detail || `HTTP error with status code ${response.status}`;
+      console.error('Server Error:', errorMsg);
+      throw new Error(errorMsg);
+    }
+
+    return json;
+  } catch (error: any) {
+    throw new HttpError(500, error.message);
+  }
+};