Baseten integration

.env.example

@@ -25,3 +25,7 @@ CARTESIA_API_KEY=your_cartesia_api_key_here
 
 # Anthropic API credentials
 ANTHROPIC_API_KEY=your_anthropic_api_key_here
+
+# Baseten API credentials
+BASETEN_API_KEY=your_baseten_api_key_here
+BASETEN_BASE_URL=your_baseten_base_url_here

agents-core/vision_agents/core/llm/llm.py

@@ -50,7 +50,6 @@ class LLM(abc.ABC):
     before_response_listener: BeforeCb
     after_response_listener: AfterCb
     agent: Optional["Agent"]
-    _conversation: Optional["Conversation"]
     function_registry: FunctionRegistry
 
     def __init__(self):
@@ -59,6 +58,8 @@ def __init__(self):
         self.events = EventManager()
         self.events.register_events_from_module(events)
         self.function_registry = FunctionRegistry()
+        self.instructions: Optional[str] = None
+        self.parsed_instructions: Optional[str] = None
 
     async def warmup(self) -> None:
         """
@@ -78,6 +79,12 @@ async def simple_response(
     ) -> LLMResponseEvent[Any]:
         raise NotImplementedError
 
+    @property
+    def _conversation(self) -> Optional[Conversation]:
+        if not self.agent:
+            return None
+        return self.agent.conversation
+
     def _build_enhanced_instructions(self) -> Optional[str]:
         """
         Build enhanced instructions by combining the original instructions with markdown file contents.
@@ -187,7 +194,6 @@ def _attach_agent(self, agent: Agent):
         Attach agent to the llm
         """
         self.agent = agent
-        self._conversation = agent.conversation
         self._set_instructions(agent.instructions)
 
     def _set_instructions(self, instructions: str):

plugins/baseten/README.md

@@ -0,0 +1,26 @@
+# Baseten Plugin for Vision Agents
+
+LLM integrations for the models hosted on Baseten for Vision Agents framework.
+
+TODO
+
+## Installation
+
+```bash
+pip install vision-agents-plugins-baseten
+```
+
+## Usage
+
+```python
+
+```
+
+
+## Requirements
+- Python 3.10+
+- `openai`
+- GetStream SDK
+
+## License
+MIT

plugins/baseten/pyproject.toml

@@ -0,0 +1,36 @@
+[build-system]
+requires = ["hatchling", "hatch-vcs"]
+build-backend = "hatchling.build"
+
+[project]
+name = "vision-agents-plugins-baseten"
+dynamic = ["version"]
+description = "Baseten plugin for vision agents"
+readme = "README.md"
+requires-python = ">=3.10"
+license = "MIT"
+dependencies = [
+    "vision-agents",
+    "openai>=2.5.0",
+]
+
+[project.urls]
+Documentation = "https://visionagents.ai/"
+Website = "https://visionagents.ai/"
+Source = "https://github.com/GetStream/Vision-Agents"
+
+[tool.hatch.version]
+source = "vcs"
+raw-options = { root = "..", search_parent_directories = true, fallback_version = "0.0.0" }
+
+[tool.hatch.build.targets.wheel]
+packages = ["."]
+
+[tool.uv.sources]
+vision-agents = { workspace = true }
+
+[dependency-groups]
+dev = [
+    "pytest>=8.4.1",
+    "pytest-asyncio>=1.0.0",
+]

plugins/baseten/vision_agents/plugins/baseten/__init__.py

@@ -0,0 +1,4 @@
+from .baseten_vlm import BasetenVLM as VLM
+
+
+__all__ = ["VLM"]