GetFileAssociations

Author: christothes

codegen/generator/src/Visitors/PaginationVisitor.cs

@@ -18,15 +18,16 @@ namespace OpenAILibraryPlugin.Visitors;
 public class PaginationVisitor : ScmLibraryVisitor
 {
 
-    private static readonly string[] _paginationParamsToReplace = ["after", "before", "limit", "order", "model", "metadata"];
+    private static readonly string[] _paginationParamsToReplace = ["after", "before", "limit", "order", "model", "metadata", "filter"];
     private static readonly Dictionary<string, string> _paramReplacementMap = new()
     {
         { "after", "AfterId" },
         { "before", "BeforeId" },
         { "limit", "PageSizeLimit" },
         { "order", "Order" },
         { "model", "Model" },
-        { "metadata", "Metadata" }
+        { "metadata", "Metadata" },
+        { "filter", "Filter" }
     };
     private static readonly Dictionary<string, (string ReturnType, string OptionsType, string[] ParamsToReplace)> _optionsReplacements = new()
     {
@@ -53,6 +54,14 @@ public class PaginationVisitor : ScmLibraryVisitor
         {
             "GetVectorStoresAsync",
             ("VectorStore", "VectorStoreCollectionOptions", _paginationParamsToReplace)
+        },
+        {
+            "GetFileAssociations",
+            ("VectorStoreFileAssociation", "VectorStoreFileAssociationCollectionOptions", _paginationParamsToReplace)
+        },
+        {
+            "GetFileAssociationsAsync",
+            ("VectorStoreFileAssociation", "VectorStoreFileAssociationCollectionOptions", _paginationParamsToReplace)
         }
     };
 

specification/base/typespec/vector-stores/models.tsp

@@ -200,9 +200,9 @@ model ListVectorStoreFilesResponse {
   // Tool customization: add a clear enum enforcement of constrained 'object' label
   object: "list";
 
-  data: VectorStoreFileObject[];
+  @pageItems data: VectorStoreFileObject[];
   first_id: string;
-  last_id: string;
+  @continuationToken last_id: string;
   has_more: boolean;
 }
 

specification/base/typespec/vector-stores/operations.tsp

@@ -126,6 +126,7 @@ interface VectorStores {
   @get
   @operationId("listVectorStoreFiles")
   @summary("Returns a list of vector store files.")
+  @list
   listVectorStoreFiles(
     ...AcceptJsonHeader,
 

specification/client/models/vector-stores.models.tsp

@@ -41,4 +41,15 @@ model VectorStoreCollectionOptions {
   ...CollectionBeforeQueryParameter,
   ...CollectionLimitQueryParameter,
   ...PageOrderQueryParameter,
+}
+
+@access(Access.public)
+@usage(Usage.input)
+model VectorStoreFileAssociationCollectionOptions {
+  ...CollectionAfterQueryParameter,
+  ...CollectionBeforeQueryParameter,
+  ...CollectionLimitQueryParameter,
+  ...PageOrderQueryParameter,
+
+  @query filter?: ListVectorStoreFilesFilter;
 }

specification/client/vector-stores.client.tsp

@@ -13,3 +13,4 @@ using Azure.ClientGenerator.Core;
 @@clientName(VectorStores.cancelVectorStoreFileBatch, "CancelBatchFileJob");
 @@clientName(VectorStores.getVectorStoreFile, "GetFileAssociation");
 @@clientName(VectorStores.listVectorStores, "GetVectorStores");
+@@clientName(VectorStores.listVectorStoreFiles, "GetFileAssociations");

src/Custom/VectorStores/Internal/Pagination/AsyncVectorStoreFileCollectionResult.cs

@@ -7,8 +7,6 @@
 
 namespace OpenAI.VectorStores;
 
-[CodeGenSuppress("GetVectorStoreFilesAsync", typeof(string), typeof(int?), typeof(string), typeof(string), typeof(string), typeof(string), typeof(RequestOptions))]
-[CodeGenSuppress("GetVectorStoreFiles", typeof(string), typeof(int?), typeof(string), typeof(string), typeof(string), typeof(string), typeof(RequestOptions))]
 [CodeGenSuppress("CreateVectorStoreFileAsync", typeof(string), typeof(BinaryContent), typeof(RequestOptions))]
 [CodeGenSuppress("CreateVectorStoreFile", typeof(string), typeof(BinaryContent), typeof(RequestOptions))]
 [CodeGenSuppress("CreateVectorStoreFileBatchAsync", typeof(string), typeof(BinaryContent), typeof(RequestOptions))]
@@ -104,77 +102,6 @@ internal virtual ClientResult GetVectorStore(string vectorStoreId, RequestOption
         return ClientResult.FromResponse(Pipeline.ProcessMessage(message, options));
     }
 
-    /// <summary>
-    /// [Protocol Method] Returns a paginated collection of vector store files.
-    /// </summary>
-    /// <param name="vectorStoreId"> The ID of the vector store that the files belong to. </param>
-    /// <param name="limit">
-    /// A limit on the number of objects to be returned. Limit can range between 1 and 100, and the
-    /// default is 20.
-    /// </param>
-    /// <param name="order">
-    /// Sort order by the `created_at` timestamp of the objects. `asc` for ascending order and`desc`
-    /// for descending order. Allowed values: "asc" | "desc"
-    /// </param>
-    /// <param name="after">
-    /// A cursor for use in pagination. `after` is an object ID that defines your place in the list.
-    /// For instance, if you make a list request and receive 100 objects, ending with obj_foo, your
-    /// subsequent call can include after=obj_foo in order to fetch the next page of the list.
-    /// </param>
-    /// <param name="before">
-    /// A cursor for use in pagination. `before` is an object ID that defines your place in the list.
-    /// For instance, if you make a list request and receive 100 objects, ending with obj_foo, your
-    /// subsequent call can include before=obj_foo in order to fetch the previous page of the list.
-    /// </param>
-    /// <param name="filter"> Filter by file status. One of `in_progress`, `completed`, `failed`, `cancelled`. </param>
-    /// <param name="options"> The request options, which can override default behaviors of the client pipeline on a per-call basis. </param>
-    /// <exception cref="ArgumentNullException"> <paramref name="vectorStoreId"/> is null. </exception>
-    /// <exception cref="ArgumentException"> <paramref name="vectorStoreId"/> is an empty string, and was expected to be non-empty. </exception>
-    /// <exception cref="ClientResultException"> Service returned a non-success status code. </exception>
-    /// <returns> A collection of service responses, each holding a page of values. </returns>
-    public virtual AsyncCollectionResult GetFileAssociationsAsync(string vectorStoreId, int? limit, string order, string after, string before, string filter, RequestOptions options)
-    {
-        Argument.AssertNotNullOrEmpty(vectorStoreId, nameof(vectorStoreId));
-
-        return new AsyncVectorStoreFileCollectionResult(this, Pipeline, options, vectorStoreId, limit, order, after, before, filter);
-    }
-
-    /// <summary>
-    /// [Protocol Method] Returns a paginated collection of vector store files.
-    /// </summary>
-    /// <param name="vectorStoreId"> The ID of the vector store that the files belong to. </param>
-    /// <param name="limit">
-    /// A limit on the number of objects to be returned. Limit can range between 1 and 100, and the
-    /// default is 20.
-    /// </param>
-    /// <param name="order">
-    /// Sort order by the `created_at` timestamp of the objects. `asc` for ascending order and`desc`
-    /// for descending order. Allowed values: "asc" | "desc"
-    /// </param>
-    /// <param name="after">
-    /// A cursor for use in pagination. `after` is an object ID that defines your place in the list.
-    /// For instance, if you make a list request and receive 100 objects, ending with obj_foo, your
-    /// subsequent call can include after=obj_foo in order to fetch the next page of the list.
-    /// </param>
-    /// <param name="before">
-    /// A cursor for use in pagination. `before` is an object ID that defines your place in the list.
-    /// For instance, if you make a list request and receive 100 objects, ending with obj_foo, your
-    /// subsequent call can include before=obj_foo in order to fetch the previous page of the list.
-    /// </param>
-    /// <param name="filter"> Filter by file status. One of `in_progress`, `completed`, `failed`, `cancelled`. </param>
-    /// <param name="options"> The request options, which can override default behaviors of the client pipeline on a per-call basis. </param>
-    /// <exception cref="ArgumentNullException"> <paramref name="vectorStoreId"/> is null. </exception>
-    /// <exception cref="ArgumentException"> <paramref name="vectorStoreId"/> is an empty string, and was expected to be non-empty. </exception>
-    /// <exception cref="ClientResultException"> Service returned a non-success status code. </exception>
-    /// <returns> A collection of service responses, each holding a page of values. </returns>
-    public virtual CollectionResult GetFileAssociations(string vectorStoreId, int? limit, string order, string after, string before, string filter, RequestOptions options)
-    {
-        Argument.AssertNotNullOrEmpty(vectorStoreId, nameof(vectorStoreId));
-
-        return new VectorStoreFileCollectionResult(this, Pipeline, options, vectorStoreId, limit, order, after, before, filter);
-    }
-
-
     /// <summary>
     /// [Protocol Method] Create a vector store file by attaching a [File](/docs/api-reference/files) to a [vector store](/docs/api-reference/vector-stores/object).
     /// </summary>