update to latest versions

Author: aaronvg

README.md

@@ -5,13 +5,17 @@ This repository contains BAML example code used in the Documentation [Tutorials]
 Clone the repo to see syntax highlights using the BAML VSCode extension!
 
 ### Requirements
+
 1. BAML CLI, BAML VSCode extension
-1. This repository uses ⚠️⚠️⚠️ Poetry ⚠️⚠️⚠️ as the python environment manager.
-2. Python >=3.9 (Contact us if you have an older version).
+2. This repository uses ⚠️⚠️⚠️ Poetry ⚠️⚠️⚠️ as the python environment manager. If you want to use Conda, pip or another dependency mgmt, run `baml init` to get yourself setup properly.
+3. Python >=3.9 (Contact us if you have an older version).
 
 **Contact us on Discord if you need help running the examples using Conda, pip or another dependency mgmt**.
 
 ### Setup
+
+Note: You can always just copy the `.baml` files you want into your own project that you have initialized with `baml init`.
+
 1. Clone the repo
 2. Install [poetry](https://python-poetry.org/docs/)
 3. Run `poetry shell` in the root
@@ -23,8 +27,9 @@ Clone the repo to see syntax highlights using the BAML VSCode extension!
 If you're having issues, reach out to us on Discord (https://discord.gg/yzaTpQ3tdT).
 
 Some common steps that help fix things:
+
 1. Make sure you also add `baml` pip package to the project dependencies if you are not using poetry (see pyproject.toml for dependency list).
 1. Make sure you're in a poetry shell when you run the python main.py files.
-2. Make sure environment variables are set (some baml files use an env.OPEN_AI_KEY). Add a .env file to the root dir, with the key set to fix this.
-3. Restart VScode if the playground isn't working
-4. Restart VSCode if you're not getting error highlights for baml-generated code, or ensure the right Python interpreter is set (Command + Shift + P -> Select interpreter)
+1. Make sure environment variables are set (some baml files use an env.OPEN_AI_KEY). Add a .env file to the root dir with the appropriate api key set to fix this.
+1. Restart VScode if the playground isn't working
+1. Restart VSCode if you're not getting error highlights for baml-generated code, or ensure the right Python interpreter is set (Command + Shift + P -> Select interpreter)

chatbot/baml_client/__do_not_import/functions/fx_classifyintent.py

@@ -8,8 +8,9 @@
 # fmt: off
 
 from ..types.enums.enm_intent import Intent
+from baml_core.stream import AsyncStream
 from baml_lib._impl.functions import BaseBAMLFunction
-from typing import List, Protocol, runtime_checkable
+from typing import AsyncIterator, Callable, List, Protocol, runtime_checkable
 
 
 IClassifyIntentOutput = List[Intent]
@@ -29,8 +30,24 @@ class IClassifyIntent(Protocol):
     async def __call__(self, *, query: str) -> List[Intent]:
         ...
 
+   
 
-class IBAMLClassifyIntent(BaseBAMLFunction[List[Intent]]):
+@runtime_checkable
+class IClassifyIntentStream(Protocol):
+    """
+    This is the interface for a stream function.
+
+    Args:
+        query: str
+
+    Returns:
+        AsyncStream[List[Intent], List[Intent]]
+    """
+
+    def __call__(self, *, query: str
+) -> AsyncStream[List[Intent], List[Intent]]:
+        ...
+class IBAMLClassifyIntent(BaseBAMLFunction[List[Intent], List[Intent]]):
     def __init__(self) -> None:
         super().__init__(
             "ClassifyIntent",
@@ -40,6 +57,10 @@ def __init__(self) -> None:
 
     async def __call__(self, *args, **kwargs) -> List[Intent]:
         return await self.get_impl("simple").run(*args, **kwargs)
+    
+    def stream(self, *args, **kwargs) -> AsyncStream[List[Intent], List[Intent]]:
+        res = self.get_impl("simple").stream(*args, **kwargs)
+        return res
 
 BAMLClassifyIntent = IBAMLClassifyIntent()
 

chatbot/baml_client/__do_not_import/functions/fx_classifyintent.pyi

@@ -8,7 +8,8 @@
 # fmt: off
 
 from ..types.enums.enm_intent import Intent
-from typing import List, Protocol, runtime_checkable
+from baml_core.stream import AsyncStream
+from typing import Callable, List, Protocol, runtime_checkable
 
 
 import typing
@@ -40,20 +41,44 @@ class IClassifyIntent(Protocol):
     async def __call__(self, *, query: str) -> List[Intent]:
         ...
 
+   
 
+@runtime_checkable
+class IClassifyIntentStream(Protocol):
+    """
+    This is the interface for a stream function.
+
+    Args:
+        query: str
+
+    Returns:
+        AsyncStream[List[Intent], List[Intent]]
+    """
+
+    def __call__(self, *, query: str
+) -> AsyncStream[List[Intent], List[Intent]]:
+        ...
 class BAMLClassifyIntentImpl:
     async def run(self, *, query: str) -> List[Intent]:
         ...
+    
+    def stream(self, *, query: str
+) -> AsyncStream[List[Intent], List[Intent]]:
+        ...
 
 class IBAMLClassifyIntent:
     def register_impl(
         self, name: ImplName
-    ) -> typing.Callable[[IClassifyIntent], IClassifyIntent]:
+    ) -> typing.Callable[[IClassifyIntent, IClassifyIntentStream], None]:
         ...
 
     async def __call__(self, *, query: str) -> List[Intent]:
         ...
 
+    def stream(self, *, query: str
+) -> AsyncStream[List[Intent], List[Intent]]:
+        ...
+
     def get_impl(self, name: ImplName) -> BAMLClassifyIntentImpl:
         ...
 
@@ -97,23 +122,34 @@ class IBAMLClassifyIntent:
         ...
 
     @typing.overload
-    def test(self, *, exclude_impl: typing.Iterable[ImplName]) -> pytest.MarkDecorator:
+    def test(self, *, exclude_impl: typing.Iterable[ImplName] = [], stream: bool = False) -> pytest.MarkDecorator:
         """
         Provides a pytest.mark.parametrize decorator to facilitate testing different implementations of
         the ClassifyIntentInterface.
 
         Args:
             exclude_impl : Iterable[ImplName]
                 The names of the implementations to exclude from testing.
+            stream: bool
+                If set, will return a streamable version of the test function.
 
         Usage:
             ```python
-            # All implementations except "simple" will be tested.
+            # All implementations except the given impl will be tested.
 
-            @baml.ClassifyIntent.test(exclude_impl=["simple"])
+            @baml.ClassifyIntent.test(exclude_impl=["implname"])
             async def test_logic(ClassifyIntentImpl: IClassifyIntent) -> None:
                 result = await ClassifyIntentImpl(...)
             ```
+
+            ```python
+            # Streamable version of the test function.
+
+            @baml.ClassifyIntent.test(stream=True)
+            async def test_logic(ClassifyIntentImpl: IClassifyIntentStream) -> None:
+                async for result in ClassifyIntentImpl(...):
+                    ...
+            ```
         """
         ...
 

chatbot/baml_client/__do_not_import/functions/fx_extractmeetingrequestinfo.py

@@ -12,8 +12,13 @@
 from ..types.classes.cls_meetingrequest import MeetingRequest
 from ..types.classes.cls_message import Message
 from ..types.enums.enm_usertype import UserType
+from ..types.partial.classes.cls_attendee import PartialAttendee
+from ..types.partial.classes.cls_conversation import PartialConversation
+from ..types.partial.classes.cls_meetingrequest import PartialMeetingRequest
+from ..types.partial.classes.cls_message import PartialMessage
+from baml_core.stream import AsyncStream
 from baml_lib._impl.functions import BaseBAMLFunction
-from typing import Protocol, runtime_checkable
+from typing import AsyncIterator, Callable, Protocol, runtime_checkable
 
 
 IExtractMeetingRequestInfoOutput = MeetingRequest
@@ -34,8 +39,25 @@ class IExtractMeetingRequestInfo(Protocol):
     async def __call__(self, *, convo: Conversation, now: str) -> MeetingRequest:
         ...
 
+   
 
-class IBAMLExtractMeetingRequestInfo(BaseBAMLFunction[MeetingRequest]):
+@runtime_checkable
+class IExtractMeetingRequestInfoStream(Protocol):
+    """
+    This is the interface for a stream function.
+
+    Args:
+        convo: Conversation
+        now: str
+
+    Returns:
+        AsyncStream[MeetingRequest, PartialMeetingRequest]
+    """
+
+    def __call__(self, *, convo: Conversation, now: str
+) -> AsyncStream[MeetingRequest, PartialMeetingRequest]:
+        ...
+class IBAMLExtractMeetingRequestInfo(BaseBAMLFunction[MeetingRequest, PartialMeetingRequest]):
     def __init__(self) -> None:
         super().__init__(
             "ExtractMeetingRequestInfo",
@@ -45,6 +67,10 @@ def __init__(self) -> None:
 
     async def __call__(self, *args, **kwargs) -> MeetingRequest:
         return await self.get_impl("simple").run(*args, **kwargs)
+    
+    def stream(self, *args, **kwargs) -> AsyncStream[MeetingRequest, PartialMeetingRequest]:
+        res = self.get_impl("simple").stream(*args, **kwargs)
+        return res
 
 BAMLExtractMeetingRequestInfo = IBAMLExtractMeetingRequestInfo()
 

chatbot/baml_client/__do_not_import/functions/fx_extractmeetingrequestinfo.pyi

@@ -12,7 +12,12 @@ from ..types.classes.cls_conversation import Conversation
 from ..types.classes.cls_meetingrequest import MeetingRequest
 from ..types.classes.cls_message import Message
 from ..types.enums.enm_usertype import UserType
-from typing import Protocol, runtime_checkable
+from ..types.partial.classes.cls_attendee import PartialAttendee
+from ..types.partial.classes.cls_conversation import PartialConversation
+from ..types.partial.classes.cls_meetingrequest import PartialMeetingRequest
+from ..types.partial.classes.cls_message import PartialMessage
+from baml_core.stream import AsyncStream
+from typing import Callable, Protocol, runtime_checkable
 
 
 import typing
@@ -45,20 +50,45 @@ class IExtractMeetingRequestInfo(Protocol):
     async def __call__(self, *, convo: Conversation, now: str) -> MeetingRequest:
         ...
 
+   
 
+@runtime_checkable
+class IExtractMeetingRequestInfoStream(Protocol):
+    """
+    This is the interface for a stream function.
+
+    Args:
+        convo: Conversation
+        now: str
+
+    Returns:
+        AsyncStream[MeetingRequest, PartialMeetingRequest]
+    """
+
+    def __call__(self, *, convo: Conversation, now: str
+) -> AsyncStream[MeetingRequest, PartialMeetingRequest]:
+        ...
 class BAMLExtractMeetingRequestInfoImpl:
     async def run(self, *, convo: Conversation, now: str) -> MeetingRequest:
         ...
+    
+    def stream(self, *, convo: Conversation, now: str
+) -> AsyncStream[MeetingRequest, PartialMeetingRequest]:
+        ...
 
 class IBAMLExtractMeetingRequestInfo:
     def register_impl(
         self, name: ImplName
-    ) -> typing.Callable[[IExtractMeetingRequestInfo], IExtractMeetingRequestInfo]:
+    ) -> typing.Callable[[IExtractMeetingRequestInfo, IExtractMeetingRequestInfoStream], None]:
         ...
 
     async def __call__(self, *, convo: Conversation, now: str) -> MeetingRequest:
         ...
 
+    def stream(self, *, convo: Conversation, now: str
+) -> AsyncStream[MeetingRequest, PartialMeetingRequest]:
+        ...
+
     def get_impl(self, name: ImplName) -> BAMLExtractMeetingRequestInfoImpl:
         ...
 
@@ -102,23 +132,34 @@ class IBAMLExtractMeetingRequestInfo:
         ...
 
     @typing.overload
-    def test(self, *, exclude_impl: typing.Iterable[ImplName]) -> pytest.MarkDecorator:
+    def test(self, *, exclude_impl: typing.Iterable[ImplName] = [], stream: bool = False) -> pytest.MarkDecorator:
         """
         Provides a pytest.mark.parametrize decorator to facilitate testing different implementations of
         the ExtractMeetingRequestInfoInterface.
 
         Args:
             exclude_impl : Iterable[ImplName]
                 The names of the implementations to exclude from testing.
+            stream: bool
+                If set, will return a streamable version of the test function.
 
         Usage:
             ```python
-            # All implementations except "simple" will be tested.
+            # All implementations except the given impl will be tested.
 
-            @baml.ExtractMeetingRequestInfo.test(exclude_impl=["simple"])
+            @baml.ExtractMeetingRequestInfo.test(exclude_impl=["implname"])
             async def test_logic(ExtractMeetingRequestInfoImpl: IExtractMeetingRequestInfo) -> None:
                 result = await ExtractMeetingRequestInfoImpl(...)
             ```
+
+            ```python
+            # Streamable version of the test function.
+
+            @baml.ExtractMeetingRequestInfo.test(stream=True)
+            async def test_logic(ExtractMeetingRequestInfoImpl: IExtractMeetingRequestInfoStream) -> None:
+                async for result in ExtractMeetingRequestInfoImpl(...):
+                    ...
+            ```
         """
         ...
 

chatbot/baml_client/__do_not_import/functions/fx_extractmeetingrequestinfopartial.py

@@ -11,8 +11,12 @@
 from ..types.classes.cls_meetingrequestpartial import MeetingRequestPartial
 from ..types.classes.cls_message import Message
 from ..types.enums.enm_usertype import UserType
+from ..types.partial.classes.cls_conversation import PartialConversation
+from ..types.partial.classes.cls_meetingrequestpartial import PartialMeetingRequestPartial
+from ..types.partial.classes.cls_message import PartialMessage
+from baml_core.stream import AsyncStream
 from baml_lib._impl.functions import BaseBAMLFunction
-from typing import Protocol, runtime_checkable
+from typing import AsyncIterator, Callable, Protocol, runtime_checkable
 
 
 IExtractMeetingRequestInfoPartialOutput = MeetingRequestPartial
@@ -33,8 +37,25 @@ class IExtractMeetingRequestInfoPartial(Protocol):
     async def __call__(self, *, convo: Conversation, now: str) -> MeetingRequestPartial:
         ...
 
+   
 
-class IBAMLExtractMeetingRequestInfoPartial(BaseBAMLFunction[MeetingRequestPartial]):
+@runtime_checkable
+class IExtractMeetingRequestInfoPartialStream(Protocol):
+    """
+    This is the interface for a stream function.
+
+    Args:
+        convo: Conversation
+        now: str
+
+    Returns:
+        AsyncStream[MeetingRequestPartial, PartialMeetingRequestPartial]
+    """
+
+    def __call__(self, *, convo: Conversation, now: str
+) -> AsyncStream[MeetingRequestPartial, PartialMeetingRequestPartial]:
+        ...
+class IBAMLExtractMeetingRequestInfoPartial(BaseBAMLFunction[MeetingRequestPartial, PartialMeetingRequestPartial]):
     def __init__(self) -> None:
         super().__init__(
             "ExtractMeetingRequestInfoPartial",
@@ -44,6 +65,10 @@ def __init__(self) -> None:
 
     async def __call__(self, *args, **kwargs) -> MeetingRequestPartial:
         return await self.get_impl("v1").run(*args, **kwargs)
+    
+    def stream(self, *args, **kwargs) -> AsyncStream[MeetingRequestPartial, PartialMeetingRequestPartial]:
+        res = self.get_impl("v1").stream(*args, **kwargs)
+        return res
 
 BAMLExtractMeetingRequestInfoPartial = IBAMLExtractMeetingRequestInfoPartial()