split wasm filter (#186)

* split wasm filter

* fix int and unit tests

* rename public_types => common and move common code there

* rename

* fix int test

crates/common/src/embeddings/create_embedding_request.rs

@@ -0,0 +1,59 @@
+/*
+ * OMF Embeddings
+ *
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 1.0.0
+ *
+ * Generated by: https://openapi-generator.tech
+ */
+
+use crate::embeddings;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
+pub struct CreateEmbeddingRequest {
+    #[serde(rename = "input")]
+    pub input: Box<embeddings::CreateEmbeddingRequestInput>,
+    /// ID of the model to use. You can use the [List models](/docs/api-reference/models/list) API to see all of your available models, or see our [Model overview](/docs/models/overview) for descriptions of them.
+    #[serde(rename = "model")]
+    pub model: String,
+    /// The format to return the embeddings in. Can be either `float` or [`base64`](https://pypi.org/project/pybase64/).
+    #[serde(rename = "encoding_format", skip_serializing_if = "Option::is_none")]
+    pub encoding_format: Option<EncodingFormat>,
+    /// The number of dimensions the resulting output embeddings should have. Only supported in `text-embedding-3` and later models.
+    #[serde(rename = "dimensions", skip_serializing_if = "Option::is_none")]
+    pub dimensions: Option<i32>,
+    /// A unique identifier representing your end-user, which can help to monitor and detect abuse. [Learn more](/docs/guides/safety-best-practices/end-user-ids).
+    #[serde(rename = "user", skip_serializing_if = "Option::is_none")]
+    pub user: Option<String>,
+}
+
+impl CreateEmbeddingRequest {
+    pub fn new(
+        input: embeddings::CreateEmbeddingRequestInput,
+        model: String,
+    ) -> CreateEmbeddingRequest {
+        CreateEmbeddingRequest {
+            input: Box::new(input),
+            model,
+            encoding_format: None,
+            dimensions: None,
+            user: None,
+        }
+    }
+}
+/// The format to return the embeddings in. Can be either `float` or [`base64`](https://pypi.org/project/pybase64/).
+#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
+pub enum EncodingFormat {
+    #[serde(rename = "float")]
+    Float,
+    #[serde(rename = "base64")]
+    Base64,
+}
+
+impl Default for EncodingFormat {
+    fn default() -> EncodingFormat {
+        Self::Float
+    }
+}

crates/common/src/embeddings/create_embedding_request_input.rs

@@ -0,0 +1,29 @@
+/*
+ * OMF Embeddings
+ *
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 1.0.0
+ *
+ * Generated by: https://openapi-generator.tech
+ */
+
+use crate::embeddings;
+use serde::{Deserialize, Serialize};
+
+/// CreateEmbeddingRequestInput : Input text to embed, encoded as a string or array of tokens. To embed multiple inputs in a single request, pass an array of strings or array of token arrays. The input must not exceed the max input tokens for the model (8192 tokens for `text-embedding-ada-002`), cannot be an empty string, and any array must be 2048 dimensions or less. for counting tokens.
+/// Input text to embed, encoded as a string or array of tokens. To embed multiple inputs in a single request, pass an array of strings or array of token arrays. The input must not exceed the max input tokens for the model (8192 tokens for `text-embedding-ada-002`), cannot be an empty string, and any array must be 2048 dimensions or less. for counting tokens.
+#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
+#[serde(untagged)]
+pub enum CreateEmbeddingRequestInput {
+    /// The string that will be turned into an embedding.
+    String(String),
+    /// The array of integers that will be turned into an embedding.
+    Array(Vec<i32>),
+}
+
+impl Default for CreateEmbeddingRequestInput {
+    fn default() -> Self {
+        Self::String(Default::default())
+    }
+}

crates/common/src/embeddings/create_embedding_response.rs

@@ -0,0 +1,55 @@
+/*
+ * OMF Embeddings
+ *
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 1.0.0
+ *
+ * Generated by: https://openapi-generator.tech
+ */
+
+use crate::embeddings;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
+pub struct CreateEmbeddingResponse {
+    /// The list of embeddings generated by the model.
+    #[serde(rename = "data")]
+    pub data: Vec<embeddings::Embedding>,
+    /// The name of the model used to generate the embedding.
+    #[serde(rename = "model")]
+    pub model: String,
+    /// The object type, which is always \"list\".
+    #[serde(rename = "object")]
+    pub object: Object,
+    #[serde(rename = "usage")]
+    pub usage: Box<embeddings::CreateEmbeddingResponseUsage>,
+}
+
+impl CreateEmbeddingResponse {
+    pub fn new(
+        data: Vec<embeddings::Embedding>,
+        model: String,
+        object: Object,
+        usage: embeddings::CreateEmbeddingResponseUsage,
+    ) -> CreateEmbeddingResponse {
+        CreateEmbeddingResponse {
+            data,
+            model,
+            object,
+            usage: Box::new(usage),
+        }
+    }
+}
+/// The object type, which is always \"list\".
+#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
+pub enum Object {
+    #[serde(rename = "list")]
+    List,
+}
+
+impl Default for Object {
+    fn default() -> Object {
+        Self::List
+    }
+}

crates/common/src/embeddings/create_embedding_response_usage.rs

@@ -0,0 +1,33 @@
+/*
+ * OMF Embeddings
+ *
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 1.0.0
+ *
+ * Generated by: https://openapi-generator.tech
+ */
+
+use crate::embeddings;
+use serde::{Deserialize, Serialize};
+
+/// CreateEmbeddingResponseUsage : The usage information for the request.
+#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
+pub struct CreateEmbeddingResponseUsage {
+    /// The number of tokens used by the prompt.
+    #[serde(rename = "prompt_tokens")]
+    pub prompt_tokens: i32,
+    /// The total number of tokens used by the request.
+    #[serde(rename = "total_tokens")]
+    pub total_tokens: i32,
+}
+
+impl CreateEmbeddingResponseUsage {
+    /// The usage information for the request.
+    pub fn new(prompt_tokens: i32, total_tokens: i32) -> CreateEmbeddingResponseUsage {
+        CreateEmbeddingResponseUsage {
+            prompt_tokens,
+            total_tokens,
+        }
+    }
+}

crates/common/src/embeddings/embedding.rs

@@ -0,0 +1,49 @@
+/*
+ * OMF Embeddings
+ *
+ * No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+ *
+ * The version of the OpenAPI document: 1.0.0
+ *
+ * Generated by: https://openapi-generator.tech
+ */
+
+use crate::embeddings;
+use serde::{Deserialize, Serialize};
+
+/// Embedding : Represents an embedding vector returned by embedding endpoint.
+#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
+pub struct Embedding {
+    /// The index of the embedding in the list of embeddings.
+    #[serde(rename = "index")]
+    pub index: i32,
+    /// The embedding vector, which is a list of floats. The length of vector depends on the model as listed in the [embedding guide](/docs/guides/embeddings).
+    #[serde(rename = "embedding")]
+    pub embedding: Vec<f64>,
+    /// The object type, which is always \"embedding\"
+    #[serde(rename = "object")]
+    pub object: Object,
+}
+
+impl Embedding {
+    /// Represents an embedding vector returned by embedding endpoint.
+    pub fn new(index: i32, embedding: Vec<f64>, object: Object) -> Embedding {
+        Embedding {
+            index,
+            embedding,
+            object,
+        }
+    }
+}
+/// The object type, which is always \"embedding\"
+#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
+pub enum Object {
+    #[serde(rename = "embedding")]
+    Embedding,
+}
+
+impl Default for Object {
+    fn default() -> Object {
+        Self::Embedding
+    }
+}

crates/common/src/embeddings/mod.rs

@@ -0,0 +1,10 @@
+pub mod create_embedding_request;
+pub use self::create_embedding_request::CreateEmbeddingRequest;
+pub mod create_embedding_request_input;
+pub use self::create_embedding_request_input::CreateEmbeddingRequestInput;
+pub mod create_embedding_response;
+pub use self::create_embedding_response::CreateEmbeddingResponse;
+pub mod create_embedding_response_usage;
+pub use self::create_embedding_response_usage::CreateEmbeddingResponseUsage;
+pub mod embedding;
+pub use self::embedding::Embedding;