fix: update context_length (#61)

Author: k223kim

requirements.txt

@@ -1 +1 @@
-lightning_sdk >= 2025.08.26
+lightning_sdk >= 2025.09.10

src/litai/llm.py

@@ -250,7 +250,9 @@ def context_length(self, model: Optional[str] = None) -> int:
         """Returns the context length of the specified model."""
         self._wait_for_model()
         assert self._llm is not None, "LLM backend must be initialized"
-        return self._llm.context_length(model)
+        if not model:
+            return self._llm.context_length
+        return self._llm.get_context_length(model)
 
     def chat(  # noqa: D417
         self,

tests/test_llm.py

@@ -92,7 +92,7 @@ def test_llm_context_length(mock_llm_class):
 
     LLMCLIENT._sdkllm_cache.clear()
     mock_llm_instance = MagicMock()
-    mock_llm_instance.context_length.return_value = 8000
+    mock_llm_instance.context_length = 8000
 
     mock_llm_class.return_value = mock_llm_instance