Support correction when creating feedback with token (#736)

Closes #735 --------- Co-authored-by: Eugene Yurtsev <[email protected]>
langchain-ai · Sep 2, 2024 · 62e648a · 62e648a
1 parent a74e072
commit 62e648a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/langserve/api_handler.py b/langserve/api_handler.py
@@ -1590,6 +1590,7 @@ async def create_feedback_from_token(
             score=create_request.score,
             value=create_request.value,
             comment=create_request.comment,
+            correction=create_request.correction,
             metadata=metadata,
         )
 

diff --git a/tests/unit_tests/test_server_client.py b/tests/unit_tests/test_server_client.py
@@ -3066,7 +3066,8 @@ async def test_token_feedback_endpoint() -> None:
             local_app, raise_app_exceptions=True
         ) as async_client:
             response = await async_client.post(
-                "/token_feedback", json={"token_or_url": "some_url", "score": 3}
+                "/token_feedback",
+                json={"token_or_url": "some_url", "score": 3},
             )
             assert response.status_code == 200
 
@@ -3077,4 +3078,5 @@ async def test_token_feedback_endpoint() -> None:
                 "metadata": {"from_langserve": True},
                 "score": 3,
                 "value": None,
+                "correction": None,
             }