answer endpoint attachments

application/api/answer/routes.py

@@ -213,13 +213,13 @@ def save_conversation(
                 "role": "assistant",
                 "content": "Summarise following conversation in no more than 3 "
                 "words, respond ONLY with the summary, use the same "
-                "language as the user query",
+                "language as the system",
             },
             {
                 "role": "user",
                 "content": "Summarise following conversation in no more than 3 words, "
                 "respond ONLY with the summary, use the same language as the "
-                "user query \n\nUser: " + question + "\n\n" + "AI: " + response,
+                "system \n\nUser: " + question + "\n\n" + "AI: " + response,
             },
         ]
 
@@ -599,6 +599,9 @@ class Answer(Resource):
             "isNoneDoc": fields.Boolean(
                 required=False, description="Flag indicating if no document is used"
             ),
+            "attachments": fields.List(
+                fields.String, required=False, description="List of attachment IDs"
+            ),
         },
     )
 
@@ -618,6 +621,7 @@ class Answer(Resource):
             )
             conversation_id = data.get("conversation_id")
             prompt_id = data.get("prompt_id", "default")
+            attachment_ids = data.get("attachments", []) 
             chunks = int(data.get("chunks", 2))
             token_limit = data.get("token_limit", settings.DEFAULT_MAX_HISTORY)
             retriever_name = data.get("retriever", "classic")
@@ -647,10 +651,14 @@ class Answer(Resource):
             if not decoded_token:
                 return make_response({"error": "Unauthorized"}, 401)
 
+            attachments = get_attachments_content(
+                attachment_ids, decoded_token.get("sub")
+            )
+
             prompt = get_prompt(prompt_id)
 
             logger.info(
-                f"/api/answer - request_data: {data}, source: {source}",
+                f"/api/answer - request_data: {data}, source: {source}, attachments: {len(attachments)}",
                 extra={"data": json.dumps({"request_data": data, "source": source})},
             )
 
@@ -664,6 +672,7 @@ class Answer(Resource):
                 prompt=prompt,
                 chat_history=history,
                 decoded_token=decoded_token,
+                attachments=attachments,
             )
 
             retriever = RetrieverCreator.create_retriever(
@@ -694,6 +703,7 @@ class Answer(Resource):
                 isNoneDoc=data.get("isNoneDoc"),
                 index=None,
                 should_save_conversation=False,
+                attachment_ids=attachment_ids,
             ):
                 try:
                     event_data = line.replace("data: ", "").strip()
@@ -744,6 +754,7 @@ class Answer(Resource):
                     llm,
                     decoded_token,
                     api_key=user_api_key,
+                    attachment_ids=attachment_ids,
                 )
             )
 

application/api/user/routes.py

@@ -42,7 +42,6 @@ from application.utils import (
     generate_image_url,
     safe_filename,
     validate_function_name,
-    validate_required_fields,
 )
 from application.vectorstore.vector_creator import VectorCreator
 
@@ -1235,6 +1234,7 @@ class CreateAgent(Resource):
         if request.content_type == "application/json":
             data = request.get_json()
         else:
+            print(request.form)
             data = request.form.to_dict()
             if "tools" in data:
                 try:
@@ -1242,18 +1242,11 @@ class CreateAgent(Resource):
                 except json.JSONDecodeError:
                     data["tools"] = []
         print(f"Received data: {data}")
-
         if data.get("status") not in ["draft", "published"]:
             return make_response(
-                jsonify(
-                    {
-                        "success": False,
-                        "message": "Status must be either 'draft' or 'published'",
-                    }
-                ),
-                400,
+                jsonify({"success": False, "message": "Invalid status"}), 400
             )
-
+        required_fields = []
         if data.get("status") == "published":
             required_fields = [
                 "name",
@@ -1264,16 +1257,11 @@ class CreateAgent(Resource):
                 "prompt_id",
                 "agent_type",
             ]
-            validate_fields = ["name", "description", "prompt_id", "agent_type"]
         else:
             required_fields = ["name"]
-            validate_fields = []
         missing_fields = check_required_fields(data, required_fields)
-        invalid_fields = validate_required_fields(data, validate_fields)
         if missing_fields:
             return missing_fields
-        if invalid_fields:
-            return invalid_fields
 
         image_url, error = handle_image_upload(request, "", user, storage)
         if error:
@@ -1282,7 +1270,7 @@ class CreateAgent(Resource):
             )
 
         try:
-            key = str(uuid.uuid4()) if data.get("status") == "published" else ""
+            key = str(uuid.uuid4())
             new_agent = {
                 "user": user,
                 "name": data.get("name"),
@@ -1467,7 +1455,6 @@ class UpdateAgent(Resource):
             return make_response(
                 jsonify({"success": False, "message": "No update data provided"}), 400
             )
-        newly_generated_key = None
         final_status = update_fields.get("status", existing_agent.get("status"))
         if final_status == "published":
             required_published_fields = [
@@ -1497,10 +1484,6 @@ class UpdateAgent(Resource):
                     ),
                     400,
                 )
-
-            if not existing_agent.get("key"):
-                newly_generated_key = str(uuid.uuid4())
-                update_fields["key"] = newly_generated_key
         update_fields["updatedAt"] = datetime.datetime.now(datetime.timezone.utc)
 
         try:
@@ -1523,7 +1506,7 @@ class UpdateAgent(Resource):
                     jsonify(
                         {
                             "success": True,
-                            "message": "Agent found, but no changes were applied",
+                            "message": "Agent found, but no changes were applied.",
                         }
                     ),
                     304,
@@ -1536,15 +1519,14 @@ class UpdateAgent(Resource):
                 jsonify({"success": False, "message": "Database error during update"}),
                 500,
             )
-        response_data = {
-            "success": True,
-            "id": agent_id,
-            "message": "Agent updated successfully",
-        }
-        if newly_generated_key:
-            response_data["key"] = newly_generated_key
         return make_response(
-            jsonify(response_data),
+            jsonify(
+                {
+                    "success": True,
+                    "id": agent_id,
+                    "message": "Agent updated successfully",
+                }
+            ),
             200,
         )
 

application/utils.py

@@ -80,7 +80,7 @@ def check_required_fields(data, required_fields):
             jsonify(
                 {
                     "success": False,
-                    "message": f"Missing required fields: {', '.join(missing_fields)}",
+                    "message": f"Missing fields: {', '.join(missing_fields)}",
                 }
             ),
             400,
@@ -88,29 +88,6 @@ def check_required_fields(data, required_fields):
     return None
 
 
-def validate_required_fields(data, required_fields):
-    missing_fields = []
-    empty_fields = []
-
-    for field in required_fields:
-        if field not in data:
-            missing_fields.append(field)
-        elif not data[field]:
-            empty_fields.append(field)
-
-    errors = []
-    if missing_fields:
-        errors.append(f"Missing required fields: {', '.join(missing_fields)}")
-    if empty_fields:
-        errors.append(f"Empty values in required fields: {', '.join(empty_fields)}")
-
-    if errors:
-        return make_response(
-            jsonify({"success": False, "message": " | ".join(errors)}), 400
-        )
-    return None
-
-
 def get_hash(data):
     return hashlib.md5(data.encode(), usedforsecurity=False).hexdigest()
 

frontend/package-lock.json

@@ -14,7 +14,6 @@
         "copy-to-clipboard": "^3.3.3",
         "i18next": "^24.2.0",
         "i18next-browser-languagedetector": "^8.0.2",
-        "lodash": "^4.17.21",
         "mermaid": "^11.6.0",
         "prop-types": "^15.8.1",
         "react": "^19.1.0",
@@ -33,7 +32,6 @@
       },
       "devDependencies": {
         "@tailwindcss/postcss": "^4.1.10",
-        "@types/lodash": "^4.17.20",
         "@types/mermaid": "^9.1.0",
         "@types/react": "^19.1.8",
         "@types/react-dom": "^19.0.0",
@@ -2304,13 +2302,6 @@
       "integrity": "sha512-HMwFiRujE5PjrgwHQ25+bsLJgowjGjm5Z8FVSf0N6PwgJrwxH0QxzHYDcKsTfV3wva0vzrpqMTJS2jXPr5BMEQ==",
       "license": "MIT"
     },
-    "node_modules/@types/lodash": {
-      "version": "4.17.20",
-      "resolved": "https://registry.npmjs.org/@types/lodash/-/lodash-4.17.20.tgz",
-      "integrity": "sha512-H3MHACvFUEiujabxhaI/ImO6gUrd8oOurg7LQtS7mbwIXA/cUqWrvBsaeJ23aZEPk1TAYkurjfMbSELfoCXlGA==",
-      "dev": true,
-      "license": "MIT"
-    },
     "node_modules/@types/mdast": {
       "version": "4.0.4",
       "resolved": "https://registry.npmjs.org/@types/mdast/-/mdast-4.0.4.tgz",
@@ -7235,12 +7226,6 @@
         "url": "https://github.com/sponsors/sindresorhus"
       }
     },
-    "node_modules/lodash": {
-      "version": "4.17.21",
-      "resolved": "https://registry.npmjs.org/lodash/-/lodash-4.17.21.tgz",
-      "integrity": "sha512-v2kDEe57lecTulaDIuNTPy3Ry4gLGJ6Z1O3vE1krgXZNrsQ+LFTGHVxVjcXPs17LhbZVGedAJv8XZ1tvj5FvSg==",
-      "license": "MIT"
-    },
     "node_modules/lodash-es": {
       "version": "4.17.21",
       "resolved": "https://registry.npmjs.org/lodash-es/-/lodash-es-4.17.21.tgz",

frontend/package.json

@@ -25,7 +25,6 @@
     "copy-to-clipboard": "^3.3.3",
     "i18next": "^24.2.0",
     "i18next-browser-languagedetector": "^8.0.2",
-    "lodash": "^4.17.21",
     "mermaid": "^11.6.0",
     "prop-types": "^15.8.1",
     "react": "^19.1.0",
@@ -44,7 +43,6 @@
   },
   "devDependencies": {
     "@tailwindcss/postcss": "^4.1.10",
-    "@types/lodash": "^4.17.20",
     "@types/mermaid": "^9.1.0",
     "@types/react": "^19.1.8",
     "@types/react-dom": "^19.0.0",

frontend/src/agents/AgentPreview.tsx

@@ -110,7 +110,7 @@ export default function AgentPreview() {
   }, [queries]);
   return (
     <div>
-      <div className="dark:bg-raisin-black flex h-full flex-col items-center justify-between gap-2 overflow-y-hidden">
+      <div className="flex h-full flex-col items-center justify-between gap-2 overflow-y-hidden dark:bg-raisin-black">
         <div className="h-[512px] w-full overflow-y-auto">
           <ConversationMessages
             handleQuestion={handleQuestion}
@@ -128,7 +128,7 @@ export default function AgentPreview() {
             showToolButton={selectedAgent ? false : true}
             autoFocus={false}
           />
-          <p className="text-gray-4000 dark:text-sonic-silver w-full self-center bg-transparent pt-2 text-center text-xs md:inline">
+          <p className="w-full self-center bg-transparent pt-2 text-center text-xs text-gray-4000 dark:text-sonic-silver md:inline">
             This is a preview of the agent. You can publish it to start using it
             in conversations.
           </p>

frontend/src/agents/NewAgent.tsx

@@ -1,4 +1,3 @@
-import isEqual from 'lodash/isEqual';
 import React, { useCallback, useEffect, useRef, useState } from 'react';
 import { useDispatch, useSelector } from 'react-redux';
 import { useNavigate, useParams } from 'react-router-dom';
@@ -9,7 +8,6 @@ import SourceIcon from '../assets/source.svg';
 import Dropdown from '../components/Dropdown';
 import { FileUpload } from '../components/FileUpload';
 import MultiSelectPopup, { OptionType } from '../components/MultiSelectPopup';
-import Spinner from '../components/Spinner';
 import AgentDetailsModal from '../modals/AgentDetailsModal';
 import ConfirmationModal from '../modals/ConfirmationModal';
 import { ActiveState, Doc, Prompt } from '../models/misc';
@@ -68,41 +66,34 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
     useState<ActiveState>('INACTIVE');
   const [agentDetails, setAgentDetails] = useState<ActiveState>('INACTIVE');
   const [addPromptModal, setAddPromptModal] = useState<ActiveState>('INACTIVE');
-  const [hasChanges, setHasChanges] = useState(false);
-  const [draftLoading, setDraftLoading] = useState(false);
-  const [publishLoading, setPublishLoading] = useState(false);
 
-  const initialAgentRef = useRef<Agent | null>(null);
   const sourceAnchorButtonRef = useRef<HTMLButtonElement>(null);
   const toolAnchorButtonRef = useRef<HTMLButtonElement>(null);
 
   const modeConfig = {
     new: {
       heading: 'New Agent',
-      buttonText: 'Publish',
+      buttonText: 'Create Agent',
       showDelete: false,
       showSaveDraft: true,
       showLogs: false,
       showAccessDetails: false,
-      trackChanges: false,
     },
     edit: {
       heading: 'Edit Agent',
-      buttonText: 'Save',
+      buttonText: 'Save Changes',
       showDelete: true,
       showSaveDraft: false,
       showLogs: true,
       showAccessDetails: true,
-      trackChanges: true,
     },
     draft: {
       heading: 'New Agent (Draft)',
-      buttonText: 'Publish',
+      buttonText: 'Publish Draft',
       showDelete: true,
       showSaveDraft: true,
       showLogs: false,
       showAccessDetails: false,
-      trackChanges: false,
     },
   };
   const chunks = ['0', '2', '4', '6', '8', '10'];
@@ -153,27 +144,23 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
     else formData.append('tools', '[]');
 
     try {
-      setDraftLoading(true);
       const response =
         effectiveMode === 'new'
           ? await userService.createAgent(formData, token)
           : await userService.updateAgent(agent.id || '', formData, token);
       if (!response.ok) throw new Error('Failed to create agent draft');
       const data = await response.json();
-
-      const updatedAgent = {
-        ...agent,
-        id: data.id || agent.id,
-        image: data.image || agent.image,
-      };
-      setAgent(updatedAgent);
-
-      if (effectiveMode === 'new') setEffectiveMode('draft');
+      if (effectiveMode === 'new') {
+        setEffectiveMode('draft');
+        setAgent((prev) => ({
+          ...prev,
+          id: data.id,
+          image: data.image || prev.image,
+        }));
+      }
     } catch (error) {
       console.error('Error saving draft:', error);
       throw new Error('Failed to save draft');
-    } finally {
-      setDraftLoading(false);
     }
   };
 
@@ -194,34 +181,26 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
     else formData.append('tools', '[]');
 
     try {
-      setPublishLoading(true);
       const response =
         effectiveMode === 'new'
           ? await userService.createAgent(formData, token)
           : await userService.updateAgent(agent.id || '', formData, token);
       if (!response.ok) throw new Error('Failed to publish agent');
       const data = await response.json();
-
-      const updatedAgent = {
-        ...agent,
-        id: data.id || agent.id,
-        key: data.key || agent.key,
-        status: 'published',
-        image: data.image || agent.image,
-      };
-      setAgent(updatedAgent);
-      initialAgentRef.current = updatedAgent;
-
+      if (data.id) setAgent((prev) => ({ ...prev, id: data.id }));
+      if (data.key) setAgent((prev) => ({ ...prev, key: data.key }));
       if (effectiveMode === 'new' || effectiveMode === 'draft') {
         setEffectiveMode('edit');
+        setAgent((prev) => ({
+          ...prev,
+          status: 'published',
+          image: data.image || prev.image,
+        }));
         setAgentDetails('ACTIVE');
       }
-      setImageFile(null);
     } catch (error) {
       console.error('Error publishing agent:', error);
       throw new Error('Failed to publish agent');
-    } finally {
-      setPublishLoading(false);
     }
   };
 
@@ -264,7 +243,6 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
         if (data.tools) setSelectedToolIds(new Set(data.tools));
         if (data.status === 'draft') setEffectiveMode('draft');
         setAgent(data);
-        initialAgentRef.current = data;
       };
       getAgent();
     }
@@ -307,19 +285,7 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
 
   useEffect(() => {
     if (isPublishable()) dispatch(setSelectedAgent(agent));
-
-    if (!modeConfig[effectiveMode].trackChanges) {
-      setHasChanges(true);
-      return;
-    }
-    if (!initialAgentRef.current) {
-      setHasChanges(false);
-      return;
-    }
-    const isChanged =
-      !isEqual(agent, initialAgentRef.current) || imageFile !== null;
-    setHasChanges(isChanged);
-  }, [agent, dispatch, effectiveMode, imageFile]);
+  }, [agent, dispatch]);
   return (
     <div className="p-4 md:p-12">
       <div className="flex items-center gap-3 px-4">
@@ -355,16 +321,10 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
           )}
           {modeConfig[effectiveMode].showSaveDraft && (
             <button
-              className="hover:bg-vi</button>olets-are-blue border-violets-are-blue text-violets-are-blue hover:bg-violets-are-blue w-28 rounded-3xl border border-solid py-2 text-sm font-medium transition-colors hover:text-white"
+              className="hover:bg-vi</button>olets-are-blue border-violets-are-blue text-violets-are-blue hover:bg-violets-are-blue rounded-3xl border border-solid px-5 py-2 text-sm font-medium transition-colors hover:text-white"
               onClick={handleSaveDraft}
             >
-              <span className="flex items-center justify-center transition-all duration-200">
-                {draftLoading ? (
-                  <Spinner size="small" color="#976af3" />
-                ) : (
-                  'Save Draft'
-                )}
-              </span>
+              Save Draft
             </button>
           )}
           {modeConfig[effectiveMode].showAccessDetails && (
@@ -385,17 +345,11 @@ export default function NewAgent({ mode }: { mode: 'new' | 'edit' | 'draft' }) {
             </button>
           )}
           <button
-            disabled={!isPublishable() || !hasChanges}
-            className={`${!isPublishable() || !hasChanges ? 'cursor-not-allowed opacity-30' : ''} bg-purple-30 hover:bg-violets-are-blue flex w-28 items-center justify-center rounded-3xl py-2 text-sm font-medium text-white`}
+            disabled={!isPublishable()}
+            className={`${!isPublishable() && 'cursor-not-allowed opacity-30'} bg-purple-30 hover:bg-violets-are-blue rounded-3xl px-5 py-2 text-sm font-medium text-white`}
             onClick={handlePublish}
           >
-            <span className="flex items-center justify-center transition-all duration-200">
-              {publishLoading ? (
-                <Spinner size="small" color="white" />
-              ) : (
-                modeConfig[effectiveMode].buttonText
-              )}
-            </span>
+            Publish
           </button>
         </div>
       </div>
@@ -700,7 +654,7 @@ function AddPromptModal({
       setNewPromptContent('');
       onSelect?.(newPromptName, newPrompt.id, newPromptContent);
     } catch (error) {
-      console.error('Error adding prompt:', error);
+      console.error(error);
     }
   };
   return (