Allow client to send steering requests to /chat/completions

aleph_alpha_client/chat.py

@@ -66,6 +66,7 @@ class ChatRequest:
     top_k: int = 0
     top_p: float = 0.0
     stream_options: Optional[StreamOptions] = None
+    steering_concepts: Optional[List[str]] = None
 
     def to_json(self) -> Mapping[str, Any]:
         payload = {k: v for k, v in asdict(self).items() if v is not None}

tests/test_chat.py

@@ -137,3 +137,26 @@ async def test_stream_options(async_client: AsyncClient, chat_model_name: str):
     # Then the last chunks has information about usage
     assert all(isinstance(item, ChatStreamChunk) for item in stream_items[:-1])
     assert isinstance(stream_items[-1], Usage)
+
+
+def test_steering_chat(sync_client: Client, chat_model_name: str):
+    base_request = ChatRequest(
+        messages=[Message(role=Role.User, content="Hello, how are you?")],
+        model="llama-3.1-8b-instruct",
+    )
+
+    steered_request = ChatRequest(
+        messages=[Message(role=Role.User, content="Hello, how are you?")],
+        model="llama-3.1-8b-instruct",
+        steering_concepts=["slang"],
+    )
+
+    base_response = sync_client.chat(base_request, model=chat_model_name)
+    steered_response = sync_client.chat(steered_request, model=chat_model_name)
+
+    base_completion_result = base_response.message.content
+    steered_completion_result = steered_response.message.content
+
+    assert base_completion_result
+    assert steered_completion_result
+    assert base_completion_result != steered_completion_result