fix(vllm): fixed vllm test for the new contexts

Author: MASATARO ASAI [email protected]

test/backends/test_vllm.py

@@ -8,7 +8,7 @@
 from mellea.backends.vllm import LocalVLLMBackend
 from mellea.backends.types import ModelOption
 import mellea.backends.model_ids as model_ids
-from mellea.stdlib.base import CBlock, LinearContext, SimpleContext
+from mellea.stdlib.base import CBlock, ChatContext, SimpleContext
 from mellea.stdlib.requirement import (
     LLMaJRequirement,
     Requirement,
@@ -36,7 +36,7 @@ def backend():
 @pytest.fixture(scope="function")
 def session(backend):
     """Fresh HuggingFace session for each test."""
-    session = MelleaSession(backend, ctx=LinearContext())
+    session = MelleaSession(backend, ctx=ChatContext())
     yield session
     session.reset()
 

test/backends/test_vllm_tools.py

@@ -7,7 +7,7 @@
 from mellea.backends.vllm import LocalVLLMBackend
 from mellea.backends.types import ModelOption
 import mellea.backends.model_ids as model_ids
-from mellea.stdlib.base import CBlock, LinearContext
+from mellea.stdlib.base import CBlock, ChatContext
 from mellea.stdlib.requirement import (
     LLMaJRequirement,
     Requirement,
@@ -34,7 +34,7 @@ def backend():
 @pytest.fixture(scope="function")
 def session(backend):
     """Fresh HuggingFace session for each test."""
-    session = MelleaSession(backend, ctx=LinearContext())
+    session = MelleaSession(backend, ctx=ChatContext())
     yield session
     session.reset()