Merge pull request #20 from airlock-protocol/worktree-agent-acc3f57a

chore: bump to v1.0.0 and make litellm optional

Author: shivdeep1

airlock/gateway/app.py

@@ -281,7 +281,7 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
     app = FastAPI(
         title="Agentic Airlock",
         description="Open agent-to-agent trust and identity verification protocol",
-        version="0.4.0",
+        version="1.0.0",
         lifespan=lifespan,
     )
 

airlock/semantic/challenge.py

@@ -18,6 +18,14 @@
 from airlock.schemas.envelope import MessageEnvelope, generate_nonce
 from airlock.schemas.identity import AgentCapability
 
+try:
+    import litellm
+
+    _HAS_LITELLM = True
+except ImportError:
+    litellm = None  # type: ignore[assignment]
+    _HAS_LITELLM = False
+
 _CONTROL_CHAR_RE = re.compile(r"[\x00-\x08\x0b\x0c\x0e-\x1f\x7f]")
 _MAX_ANSWER_LENGTH = 2000
 
@@ -347,9 +355,11 @@ async def _generate_question(
 
     prompt = _GENERATION_PROMPT.format(capabilities=cap_text)
 
-    try:
-        import litellm
+    if not _HAS_LITELLM:
+        logger.info("litellm not installed, using fallback question")
+        return _select_fallback_question(session_id, capabilities)
 
+    try:
         kwargs: dict[str, Any] = {
             "model": model,
             "messages": [{"role": "user", "content": prompt}],
@@ -358,7 +368,7 @@ async def _generate_question(
         if api_base:
             kwargs["api_base"] = api_base
 
-        response = await asyncio.wait_for(litellm.acompletion(**kwargs), timeout=30)
+        response = await asyncio.wait_for(litellm.acompletion(**kwargs), timeout=30)  # type: ignore[union-attr]
         raw = response.choices[0].message.content
         question = (raw or "").strip()
         if question:
@@ -547,9 +557,10 @@ async def _evaluate_with_llm(
     prompt_template = _EVALUATION_PROMPT_STRUCTURED if use_structured else _EVALUATION_PROMPT
     prompt = prompt_template.format(question=question, answer=answer)
 
-    try:
-        import litellm
+    if not _HAS_LITELLM:
+        return ChallengeOutcome.AMBIGUOUS, "LLM evaluation unavailable (litellm not installed)"
 
+    try:
         kwargs: dict[str, Any] = {
             "model": model,
             "messages": [{"role": "user", "content": prompt}],
@@ -562,7 +573,7 @@ async def _evaluate_with_llm(
         if use_structured:
             kwargs["response_format"] = {"type": "json_object"}
 
-        response = await asyncio.wait_for(litellm.acompletion(**kwargs), timeout=30)
+        response = await asyncio.wait_for(litellm.acompletion(**kwargs), timeout=30)  # type: ignore[union-attr]
         raw = response.choices[0].message.content
         content = (raw or "").strip()
         if not content:

pyproject.toml

@@ -10,7 +10,7 @@ packages = ["airlock"]
 
 [project]
 name = "airlock-protocol"
-version = "0.4.0"
+version = "1.0.0"
 description = "An open protocol for agent-to-agent trust verification"
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -39,7 +39,6 @@ dependencies = [
     "httpx>=0.27.0",
     "langgraph>=0.2.0,<1.0",
     "lancedb>=0.17.0,<1.0",
-    "litellm>=1.50.0,<2.0",
     "pyarrow>=17.0.0",
     "PyJWT>=2.8.0,<3.0",
     "click>=8.1.0,<9.0",
@@ -81,6 +80,9 @@ openai = [
 anthropic = [
     "anthropic>=0.30.0",
 ]
+llm = [
+    "litellm>=1.50.0,<2.0",
+]
 
 [tool.ruff]
 target-version = "py311"

tests/test_litellm_optional.py

@@ -0,0 +1,17 @@
+"""Verify app works without litellm installed."""
+from __future__ import annotations
+
+from airlock.config import AirlockConfig
+
+
+class TestLitellmOptional:
+    def test_config_loads(self) -> None:
+        cfg = AirlockConfig()
+        assert cfg.env == "development"
+
+    def test_app_creates(self) -> None:
+        from airlock.gateway.app import create_app
+
+        app = create_app()
+        assert app is not None
+        assert app.version == "1.0.0"