Add struct_name to Client

Author: emilsoman

README.md

@@ -65,7 +65,8 @@ Now call the BAML function:
 ```elixir
 # from: The path to the baml_src directory.
 # namespace: The module name under which the returned structs will be nested.
-%BamlElixir.Client{from: "priv/baml_src", namespace: "MyApp.BamlClient"}
+# struct_name: The module name which will be used for the returned struct.
+%BamlElixir.Client{from: "priv/baml_src", namespace: "MyApp.BamlClient", struct_name: MyResumeStruct}
 |> BamlElixir.Native.call(c, "ExtractResume", %{resume: "John Doe is the CTO of Acme Inc."})
 ```
 

lib/baml_elixir/client.ex

@@ -1,4 +1,25 @@
 defmodule BamlElixir.Client do
-  defstruct namespace: "BamlClient",
-            from: "baml_src"
+  defstruct namespace: nil,
+            from: "baml_src",
+            struct_name: nil
+
+  def call(client, function_name, args) do
+    client = %{
+      client
+      | namespace: client.namespace || "",
+        struct_name: struct_name(client.struct_name)
+    }
+
+    BamlElixir.Native.call(client, function_name, args)
+  end
+
+  defp struct_name(struct_name) do
+    if struct_name do
+      struct_name
+      |> Module.split()
+      |> Enum.join(".")
+    else
+      ""
+    end
+  end
 end

mix.exs

@@ -1,7 +1,7 @@
 defmodule BamlElixir.MixProject do
   use Mix.Project
 
-  @version "0.1.0"
+  @version "0.2.0"
 
   def project do
     [

native/baml_elixir/src/lib.rs

@@ -97,8 +97,19 @@ fn baml_value_to_term<'a>(env: Env<'a>, value: &BamlValue, client: &Client) -> N
             Ok(result_map)
         }
         BamlValue::Class(class_name, fields) => {
-            // Create an Elixir struct for the class using the client's namespace
-            let module_name = format!("Elixir.{}.{}", client.namespace, class_name);
+            // Create an Elixir struct for the class using the client's namespace and struct_name
+            let struct_or_class = if client.struct_name.is_empty() {
+                class_name
+            } else {
+                &client.struct_name
+            };
+
+            let module_name = if !client.namespace.is_empty() {
+                format!("Elixir.{}.{}", client.namespace, struct_or_class)
+            } else {
+                format!("Elixir.{}", struct_or_class)
+            };
+
             let mut struct_term = elixir_struct::make_ex_struct(env, &module_name)
                 .map_err(|_| Error::Term(Box::new("Failed to create struct")))?;
             // Add all fields
@@ -116,6 +127,7 @@ fn baml_value_to_term<'a>(env: Env<'a>, value: &BamlValue, client: &Client) -> N
 #[module = "BamlElixir.Client"]
 struct Client {
     namespace: String,
+    struct_name: String,
     from: String,
 }