fix: clean up code

akai-shuuichi · Nov 11, 2024 · cfff8ee · cfff8ee
1 parent f49f1a0
commit cfff8ee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lmdeploy/serve/async_engine.py b/lmdeploy/serve/async_engine.py
@@ -660,7 +660,8 @@ def parse_tool_response(self, text, tools, **kwargs):
             else:
                 text, action = action
             action = json.loads(action)
-            name, parameters = action['name'], json.dumps(action.get('parameters', action.get('arguments', {})))
+            name, parameters = action['name'], json.dumps(
+                action.get('parameters', action.get('arguments', {})))
         else:
             raise RuntimeError(f'Unexpected model response: {text}')
         action_id = [tool.function.name for tool in tools].index(name)