patch: Use correct attribute for token url (#662)

Use correct attribute
langchain-ai · May 24, 2024 · 81a633e · 81a633e
1 parent bb74431
commit 81a633e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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",
                                 }
                             ]
                         },