fix: custom tool input number fail (#6200)

Co-authored-by: jinqi.guo <[email protected]>
langgenius · Jul 14, 2024 · 4fdcb30 · 4fdcb30
1 parent 07add06
commit 4fdcb30
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/api/core/app/apps/advanced_chat/app_runner.py b/api/core/app/apps/advanced_chat/app_runner.py
@@ -255,6 +255,12 @@ def _stream_output(self, queue_manager: AppQueueManager,
                 )
                 index += 1
                 time.sleep(0.01)
+        else:
+            queue_manager.publish(
+                QueueTextChunkEvent(
+                    text=text
+                ), PublishFrom.APPLICATION_MANAGER
+            )
 
         queue_manager.publish(
             QueueStopEvent(stopped_by=stopped_by),

diff --git a/api/core/tools/tool/api_tool.py b/api/core/tools/tool/api_tool.py
@@ -238,7 +238,7 @@ def _convert_body_property_type(self, property: dict[str, Any], value: Any) -> A
                     return int(value)
                 elif property['type'] == 'number':
                     # check if it is a float
-                    if '.' in value:
+                    if '.' in str(value):
                         return float(value)
                     else:
                         return int(value)