[bug] Plumb type-builder to streaming during tests (#1660)

Type builder wasn't pushed down to streaming which caused
deserialization errors.

<!-- ELLIPSIS_HIDDEN -->


----

> [!IMPORTANT]
> Fixes deserialization errors by ensuring type builder is passed to
streaming in `BamlRuntime`, adds `ExtractEntitiesTest`, and updates
`Ollama` client model.
> 
>   - **Behavior**:
> - Fixes deserialization errors by ensuring type builder is passed to
streaming in `BamlRuntime` in `lib.rs`.
>     - Updates `stream.run()` call to include `type_builder`.
>   - **Tests**:
> - Adds `ExtractEntitiesTest` in `dynamic_tests.baml` to verify type
builder integration.
>     - Introduces `DynamicSchema` class for testing purposes.
>   - **Models**:
> - Updates `Ollama` client model from `llama2` to `llama3.1` in
`clients.baml`.
> 
> <sup>This description was created by </sup>[<img alt="Ellipsis"
src="https://img.shields.io/badge/Ellipsis-blue?color=175173">](https://www.ellipsis.dev?ref=BoundaryML%2Fbaml&utm_source=github&utm_medium=referral)<sup>
for e5999eb. It will automatically
update as commits are pushed.</sup>


<!-- ELLIPSIS_HIDDEN -->

Author: hellovai

engine/baml-runtime/src/lib.rs

@@ -293,19 +293,18 @@ impl BamlRuntime {
                 ctx.create_ctx(type_builder.as_ref(), None, span.clone().map(|s| s.span_id))?;
             let (params, constraints) =
                 self.get_test_params_and_constraints(function_name, test_name, &rctx, true)?;
-            let rctx_stream =
-                ctx.create_ctx(type_builder.as_ref(), None, span.clone().map(|s| s.span_id))?;
             let mut stream = self.inner.stream_function_impl(
                 function_name.into(),
                 &params,
                 self.tracer.clone(),
-                rctx_stream,
+                rctx,
                 #[cfg(not(target_arch = "wasm32"))]
                 self.async_runtime.clone(),
                 // TODO: collectors here?
                 vec![],
             )?;
-            let (response_res, span_uuid) = stream.run(on_event, ctx, None, None).await;
+            let (response_res, span_uuid) =
+                stream.run(on_event, ctx, type_builder.as_ref(), None).await;
             let res = response_res?;
             let (_, llm_resp, val) = res
                 .event_chain()

integ-tests/baml_src/clients.baml

@@ -68,7 +68,7 @@ client<llm> GPT35LegacyProvider {
 client<llm> Ollama {
   provider ollama
   options {
-    model llama2
+    model llama3.1
   }
 }
 

integ-tests/baml_src/test-files/dynamic/dynamic_tests.baml

@@ -0,0 +1,57 @@
+class DynamicSchema {
+  @@dynamic
+}
+
+function ExtractEntities(text: string) -> DynamicSchema {
+  client Ollama
+  prompt #"
+    Parse the following text and return a structured representation of the data in the schema below.
+
+    text:
+    ---
+    {{text}}
+    ---
+
+    {{ ctx.output_format }}
+  "#
+}
+
+
+test ExtractEntitiesTest {
+  functions [ExtractEntities]
+  
+  type_builder {
+        enum Gender {
+            MALE
+            FEMALE
+            OTHER
+        }
+        enum PreferredContactInfo{
+            EMAIL
+            PHONE
+        }
+        class PersonalInfo{
+            
+            first_name  string?
+            last_name  string?
+            gender  Gender?
+            
+        }
+        class ContactInfo2 {
+            email  string? @description(#"The customer's email address"#)
+            phone  string? @description(#"The customer's phone number"#)  
+            preferred_contact_method  PreferredContactInfo? @description(#"The customer's preferred method of contact"#)   
+
+        }
+        dynamic class DynamicSchema {
+            personal_info PersonalInfo
+            contact_info ContactInfo2
+        }
+  }
+  
+  args {    
+       "text" "Agent: Good morning! Thank you for reaching out. I’ll need to collect some basic details to assist you better. Could you please provide your first and last name? Customer: Sure! My name is John Doe. Agent: Thank you, John. May I also ask for your gender? Customer: I'd prefer not to share that at the moment. Agent: No problem at all. Now, for contact purposes, could you share your email address? Customer: Yes, my email is johndo@example.com. Agent: Great! Do you have a phone number where we can reach you? Customer: I’d rather not provide that right now. Agent: That’s completely fine. How would you prefer us to contact you—by email or phone? Customer: Please contact me via Email. Agent: Understood! Lastly, can you share the reason for your call today? Customer: I’m not ready to specify that just yet. Agent: That’s okay, John! I’ve noted everything down. If you need any further assistance, feel free to reach out. Have a great day!"
+  } 
+   
+  
+}