chore: disable zod validation for integration testing

paisley-rag · Aug 7, 2024 · 635da1c · 635da1c
1 parent 0b1f376
commit 635da1c
Show file tree

Hide file tree

Showing 5 changed files with 55 additions and 72 deletions.
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/pipeline/config_util.py b/pipeline/config_util.py
@@ -30,7 +30,7 @@ def ui_to_pipeline(ui_json):
         },
         'generative_model': ui_obj['generative_model'],
         'prompt': {
-            'on': 'True' if ui_obj.get('prompt') else 'False',
+            'on': ui_obj.get('prompt', False),
             'template_str': ui_obj.get('prompt', '')
         }
     }

diff --git a/routers/chatbots.py b/routers/chatbots.py
@@ -29,7 +29,7 @@ async def get_chatbots():
     log.info('/api/chatbots loaded')
     results = mutil.get_all(CONFIG_DB, CONFIG_PIPELINE_COL, {}, { '_id': 0 })
     log.info('/api/chatbots results:', results)
-    return json.dumps(results)
+    return results
 
 @router.get('/{id}')
 async def get_chatbots_id(id: str):
@@ -38,7 +38,7 @@ async def get_chatbots_id(id: str):
     if not results:
         return json.dumps({"message": "no chatbot configuration found"})
 
-    return json.dumps(results)
+    return results
 
 @router.post('/')
 async def post_chatbots(request: Request):

diff --git a/ui/src/services/chatbot-service.ts b/ui/src/services/chatbot-service.ts
@@ -50,57 +50,19 @@ export type ServerPipelineConfig = z.infer<typeof serverPipelineConfigSchema>;
 
 async function updateChatbot(id: string, data: ClientPipelineConfig) {
   const response = await axios.put(`${baseUrl}/api/chatbots/${id}`, data);
-  return serverPipelineConfigSchema.parse(JSON.parse(response.data));
+  return response.data;
 }
 
 async function fetchChatbots() {
   const response = await axios.get(`${baseUrl}/api/chatbots`);
-  const chatbots = serverPipelinesConfigSchema.parse(JSON.parse(response.data));
-  const clientChatbots: ClientPipelineConfig[] = chatbots.map((chatbot) => ({
-    id: chatbot.id,
-    name: chatbot.name,
-    knowledge_bases: chatbot.knowledgebases,
-    generative_model: chatbot.generative_model,
-    similarity: {
-      on: chatbot.postprocessing.similarity.on === "True",
-      cutoff: chatbot.postprocessing.similarity.cutoff,
-    },
-    colbert_rerank: {
-      on: chatbot.postprocessing.colbertRerank.on === "True",
-      top_n: chatbot.postprocessing.colbertRerank.top_n,
-    },
-    long_context_reorder: {
-      on: chatbot.postprocessing.longContextReorder.on === "True",
-    },
-    prompt: "",
-  }));
-  return clientChatbots;
+  console.log(response.data);
+  return response.data;
 }
 
 async function fetchChatbotById(id: string) {
   const response = await axios.get(`${baseUrl}/api/chatbots/${id}`);
-  const chatbot = serverPipelineConfigSchema.parse(JSON.parse(response.data));
-  console.log(chatbot);
-  const clientChatbot: ClientPipelineConfig = {
-    id: chatbot.id,
-    name: chatbot.name,
-    knowledge_bases: chatbot.knowledgebases,
-    generative_model: chatbot.generative_model,
-    similarity: {
-      on: chatbot.postprocessing.similarity.on === "True",
-      cutoff: chatbot.postprocessing.similarity.cutoff,
-    },
-    colbert_rerank: {
-      on: chatbot.postprocessing.colbertRerank.on === "True",
-      top_n: chatbot.postprocessing.colbertRerank.top_n,
-    },
-    long_context_reorder: {
-      on: chatbot.postprocessing.longContextReorder.on === "True",
-    },
-    prompt: "",
-  };
-
-  return clientChatbot;
+  console.log(response.data);
+  return response.data;
 }
 
 export const chatbotService = {

diff --git a/ui/src/services/knowledge-base-service.ts b/ui/src/services/knowledge-base-service.ts
@@ -120,17 +120,20 @@ export type ServerKnowledgeBaseConfig = z.infer<
 
 async function fetchKnowledgeBases() {
   const response = await axios.get(`${baseUrl}/api/knowledge-bases`);
-  return knowledgeBasesSchema.parse(response.data);
+  console.log(response.data);
+  return response.data;
 }
 
 async function fetchKnowledgeBaseById(id: string) {
   const response = await axios.get(`${baseUrl}/api/knowledge-base/${id}`);
-  return serverKnowledgeBaseConfigSchema.parse(response.data);
+  console.log(response.data);
+  return response.data;
 }
 
 async function createKnowledgeBase(config: ClientKnowledgeBaseConfig) {
   const response = await axios.post(`${baseUrl}/api/knowledge-bases`, config);
-  return serverKnowledgeBaseConfigSchema.parse(response.data);
+  console.log(response.data);
+  return response.data;
 }
 
 async function uploadFile(id: string, file: File) {