feat(integrations): add typespec for filename argument

julep-ai · Nov 9, 2024 · a0781c8 · a0781c8
1 parent 8dbf489
commit a0781c8
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 2 deletions.
diff --git a/agents-api/agents_api/autogen/Tools.py b/agents-api/agents_api/autogen/Tools.py
@@ -983,6 +983,10 @@ class LlamaParseFetchArguments(BaseModel):
     """
     Language of the text. Default is English.
     """
+    filename: str | None = None
+    """
+    File Name. If not provided, a random name will be generated.
+    """
     file: str
     """
     The base64 string of the file
@@ -1013,6 +1017,10 @@ class LlamaParseFetchArgumentsUpdate(BaseModel):
     """
     Language of the text. Default is English.
     """
+    filename: str | None = None
+    """
+    File Name. If not provided, a random name will be generated.
+    """
     file: str | None = None
     """
     The base64 string of the file

diff --git a/integrations-service/integrations/autogen/Tools.py b/integrations-service/integrations/autogen/Tools.py
@@ -983,6 +983,10 @@ class LlamaParseFetchArguments(BaseModel):
     """
     Language of the text. Default is English.
     """
+    filename: str | None = None
+    """
+    File Name. If not provided, a random name will be generated.
+    """
     file: str
     """
     The base64 string of the file
@@ -1013,6 +1017,10 @@ class LlamaParseFetchArgumentsUpdate(BaseModel):
     """
     Language of the text. Default is English.
     """
+    filename: str | None = None
+    """
+    File Name. If not provided, a random name will be generated.
+    """
     file: str | None = None
     """
     The base64 string of the file

diff --git a/integrations-service/integrations/utils/integrations/llama_parse.py b/integrations-service/integrations/utils/integrations/llama_parse.py
@@ -34,8 +34,8 @@ async def parse(
     # Decode base64 file content
     file_content = base64.b64decode(arguments.file)
     extra_info = {
-        "file_name": str(uuid.uuid4())
-    }  # Generate a random UUID for the filename
+        "file_name": arguments.filename if arguments.filename else str(uuid.uuid4())
+    }
 
     # Parse the document
     documents = await parser.aload_data(file_content, extra_info=extra_info)

diff --git a/typespec/tools/llama_parse.tsp b/typespec/tools/llama_parse.tsp
@@ -26,6 +26,9 @@ model LlamaParseFetchArguments {
     /** Language of the text. Default is English.*/
     language?: string = "en";
 
+    /** File Name. If not provided, a random name will be generated. */
+    filename?: string;
+
     /** The base64 string of the file*/
     file: string;
 }

diff --git a/typespec/tsp-output/@typespec/openapi3/openapi-1.0.0.yaml b/typespec/tsp-output/@typespec/openapi3/openapi-1.0.0.yaml
@@ -7086,6 +7086,9 @@ components:
           type: string
           description: Language of the text. Default is English.
           default: en
+        filename:
+          type: string
+          description: File Name. If not provided, a random name will be generated.
         file:
           type: string
           description: The base64 string of the file
@@ -7115,6 +7118,9 @@ components:
           type: string
           description: Language of the text. Default is English.
           default: en
+        filename:
+          type: string
+          description: File Name. If not provided, a random name will be generated.
         file:
           type: string
           description: The base64 string of the file