add support for "configurable" langserve argument

sunholo-data · May 13, 2024 · 44c01ae · 44c01ae
1 parent dacac2d
commit 44c01ae
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.52.1'
+version = '0.52.2'
 
 setup(
     name='sunholo',

diff --git a/sunholo/agents/langserve.py b/sunholo/agents/langserve.py
@@ -35,10 +35,18 @@ def prepare_request_data(user_input, endpoint, vector_name, **kwargs):
         key = next(iter(input_schema['properties']))
         input_data = {key: user_input}
 
+        # Check for the special 'configurable' key in kwargs
+        config_data = {}
+        if 'configurable' in kwargs:
+            config_data['configurable'] = kwargs.pop('configurable')
+
         # Merge kwargs into request_data
         input_data.update(kwargs)
         input_data['vector_name'] = vector_name
         request_data = {"input": input_data}
+        if config_data:
+            request_data['config'] = config_data
+
         return request_data
     else:
         log.error("Invalid or no input schema available.")