diff --git a/langserve/api_handler.py b/langserve/api_handler.py index 346e89dc..d6b24b58 100644 --- a/langserve/api_handler.py +++ b/langserve/api_handler.py @@ -852,7 +852,7 @@ async def invoke( feedback_tokens=[ FeedbackToken( key=feedback_key, - url=feedback_token.url, + token_url=feedback_token.url, expires_at=feedback_token.expires_at.isoformat(), ) ] @@ -999,7 +999,7 @@ async def batch( feedback_tokens=[ FeedbackToken( key=feedback_key, - url=feedback_token.url, + token_url=feedback_token.url, expires_at=feedback_token.expires_at.isoformat(), ) ], diff --git a/tests/unit_tests/test_server_client.py b/tests/unit_tests/test_server_client.py index ff6089e1..92c7118e 100644 --- a/tests/unit_tests/test_server_client.py +++ b/tests/unit_tests/test_server_client.py @@ -2853,7 +2853,7 @@ async def test_token_feedback_included_in_responses() -> None: { "expires_at": "2023-01-01T00:00:00", "key": "foo", - "token_url": None, + "token_url": "feedback_id", } ], "run_id": run_id, @@ -2883,7 +2883,7 @@ async def test_token_feedback_included_in_responses() -> None: { "expires_at": "2023-01-01T00:00:00", "key": "foo", - "token_url": None, + "token_url": "feedback_id", } ] }, @@ -2892,7 +2892,7 @@ async def test_token_feedback_included_in_responses() -> None: { "expires_at": "2023-01-01T00:00:00", "key": "foo", - "token_url": None, + "token_url": "feedback_id", } ] },