Adds support for Anthropic models (#72)

* Adds support for Anthropic's Claude models

* update anthropic env var in test.yml

* remove condition since model is validated already

* move comment to end

* fix Anthropic API tests to use 'thoughts' field instead of 'result'

* add integration tests

* restore test files

* add test for anthropic in microbot test

---------

Co-authored-by: Kavya Sree Kaitepalli <kkaitepalli@microsoft.com>

Author: KavyaSree2610

.github/workflows/test.yml

@@ -111,6 +111,10 @@ jobs:
           AZURE_OPENAI_ENDPOINT: ${{ secrets.AZURE_OPENAI_ENDPOINT }}
           BROWSER_USE_LLM_MODEL: "gpt-5"
           BROWSER_USE_LLM_TEMPERATURE: 1
+          #Anthrpic API Configuration
+          ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}
+          ANTHROPIC_DEPLYMENT_NAME: ${{ secrets.ANTHROPIC_DEPLOYMENT_NAME }}
+          ANTHROPIC_END_POINT: ${{ secrets.ANTHROPIC_END_POINT }}
         run: |
           python -m pytest ${{ matrix.pytest-args }} \
             -n auto \

requirements.txt

@@ -1,4 +1,5 @@
 openai==1.107.3
+anthropic==0.75.0
 python-dotenv==1.1.1
 docker==7.1.0
 fastapi==0.116.1

src/microbots/MicroBot.py

@@ -10,6 +10,7 @@
 from microbots.environment.local_docker.LocalDockerEnvironment import (
     LocalDockerEnvironment,
 )
+from microbots.llm.anthropic_api import AnthropicApi
 from microbots.llm.openai_api import OpenAIApi
 from microbots.llm.ollama_local import OllamaLocal
 from microbots.llm.llm import llm_output_format_str
@@ -279,6 +280,10 @@ def _create_llm(self):
             self.llm = OllamaLocal(
                 system_prompt=self.system_prompt, model_name=self.deployment_name
             )
+        elif self.model_provider == ModelProvider.ANTHROPIC:
+            self.llm = AnthropicApi(
+                system_prompt=self.system_prompt, deployment_name=self.deployment_name
+            )
         # No Else case required as model provider is already validated using _validate_model_and_provider
 
     def _validate_model_and_provider(self, model):

src/microbots/constants.py

@@ -5,6 +5,7 @@
 class ModelProvider(StrEnum):
     OPENAI = "azure-openai"
     OLLAMA_LOCAL = "ollama-local"
+    ANTHROPIC = "anthropic"
 
 
 class ModelEnum(StrEnum):

src/microbots/llm/anthropic_api.py

@@ -0,0 +1,66 @@
+import json
+import os
+from dataclasses import asdict
+from logging import getLogger
+
+from dotenv import load_dotenv
+from anthropic import Anthropic
+from microbots.llm.llm import LLMAskResponse, LLMInterface
+
+logger = getLogger(__name__)
+
+load_dotenv()
+
+endpoint = os.getenv("ANTHROPIC_END_POINT")
+deployment_name = os.getenv("ANTHROPIC_DEPLOYMENT_NAME")
+api_key = os.getenv("ANTHROPIC_API_KEY")
+
+
+class AnthropicApi(LLMInterface):
+
+    def __init__(self, system_prompt, deployment_name=deployment_name, max_retries=3):
+        self.ai_client = Anthropic(
+            api_key=api_key,
+            base_url=endpoint
+        )
+        self.deployment_name = deployment_name
+        self.system_prompt = system_prompt
+        self.messages = []
+
+        # Set these values here. This logic will be handled in the parent class.
+        self.max_retries = max_retries
+        self.retries = 0
+
+    def ask(self, message) -> LLMAskResponse:
+        self.retries = 0  # reset retries for each ask. Handled in parent class.
+
+        self.messages.append({"role": "user", "content": message})
+
+        valid = False
+        while not valid:
+            response = self.ai_client.messages.create(
+                model=self.deployment_name,
+                system=self.system_prompt,
+                messages=self.messages,
+                max_tokens=4096,
+            )
+            
+            # Extract text content from response
+            response_text = response.content[0].text if response.content else ""
+            logger.debug("Raw Anthropic response (first 500 chars): %s", response_text[:500])
+            
+            # Try to extract JSON if wrapped in markdown code blocks
+            import re
+            json_match = re.search(r'```(?:json)?\s*(\{.*?\})\s*```', response_text, re.DOTALL)
+            if json_match:
+                response_text = json_match.group(1)
+            
+            valid, askResponse = self._validate_llm_response(response=response_text)
+
+        self.messages.append({"role": "assistant", "content": json.dumps(asdict(askResponse))})
+
+        return askResponse
+
+    def clear_history(self):
+        self.messages = []
+        return True

test/bot/test_microbot.py

@@ -8,6 +8,7 @@
 from pathlib import Path
 import subprocess
 import sys
+from unittest.mock import patch, Mock
 
 import pytest
 # Add src directory to path to import from local source
@@ -79,6 +80,19 @@ def no_mount_microBot(self):
         yield bot
         del bot
 
+    @pytest.fixture(scope="function")
+    def anthropic_microBot(self):
+        with patch('microbots.llm.anthropic_api.endpoint', 'https://api.anthropic.com'), \
+             patch('microbots.llm.anthropic_api.deployment_name', 'claude-sonnet-4'), \
+             patch('microbots.llm.anthropic_api.api_key', 'test-api-key'), \
+             patch('microbots.llm.anthropic_api.Anthropic'):
+            bot = MicroBot(
+                model="anthropic/claude-sonnet-4",
+                system_prompt=SYSTEM_PROMPT,
+            )
+            yield bot
+            del bot
+
     def test_microbot_ro_mount(self, ro_microBot, test_repo: Path):
         assert test_repo is not None
 
@@ -100,6 +114,15 @@ def test_microbot_overlay_teardown(self, ro_microBot, caplog):
 
         assert "Failed to remove working directory" not in caplog.text
 
+    def test_microbot_anthropic_initialization(self, anthropic_microBot):
+        """Test that MicroBot correctly initializes with Anthropic model provider."""
+        assert anthropic_microBot is not None
+        assert anthropic_microBot.model_provider == "anthropic"
+        assert anthropic_microBot.deployment_name == "claude-sonnet-4"
+        assert anthropic_microBot.llm is not None
+        from microbots.llm.anthropic_api import AnthropicApi
+        assert isinstance(anthropic_microBot.llm, AnthropicApi)
+
     def test_microbot_2bot_combo(self, log_file_path, test_repo, issue_1):
         assert test_repo is not None
         assert log_file_path is not None
@@ -229,6 +252,28 @@ def test_incorrect_model_format(self, test_repo):
                 folder_to_mount=test_repo_mount_ro,
             )
 
+    def test_microbot_anthropic_with_mount(self, test_repo):
+        """Test that MicroBot with Anthropic provider works with mounted folders."""
+        assert test_repo is not None
+
+        test_repo_mount_ro = Mount(
+            str(test_repo), f"{DOCKER_WORKING_DIR}/{test_repo.name}", PermissionLabels.READ_ONLY
+        )
+        with patch('microbots.llm.anthropic_api.endpoint', 'https://api.anthropic.com'), \
+             patch('microbots.llm.anthropic_api.deployment_name', 'claude-sonnet-4'), \
+             patch('microbots.llm.anthropic_api.api_key', 'test-api-key'), \
+             patch('microbots.llm.anthropic_api.Anthropic'):
+            bot = MicroBot(
+                model="anthropic/claude-sonnet-4",
+                system_prompt=SYSTEM_PROMPT,
+                folder_to_mount=test_repo_mount_ro,
+            )
+            assert bot is not None
+            assert bot.model_provider == "anthropic"
+            from microbots.llm.anthropic_api import AnthropicApi
+            assert isinstance(bot.llm, AnthropicApi)
+            del bot
+
     def test_max_iterations_exceeded(self, no_mount_microBot, monkeypatch):
         assert no_mount_microBot is not None
 

test/llm/conftest.py

@@ -336,9 +336,13 @@ def test_ollama_response_parsing(mock_ollama_response):
     }
 
 
-# Marker for tests that require Ollama Local
+# Marker for tests that require Ollama Local and anthropic model
 def pytest_configure(config):
     config.addinivalue_line(
         "markers",
         "ollama_local: mark test as requiring Ollama Local setup (deselect with '-m \"not ollama_local\"')"
     )
+    config.addinivalue_line(
+        "markers",
+        "anthropic_integration: mark test as requiring Anthropic API (deselect with '-m \"not anthropic_integration\"')"
+    )