
    i                        d dl Z d dlmZmZmZmZmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ d dlmZ  e       Zej%                  d       ed	       ed	       ed	       ed
       ed      fdedededededz  f
d       Zy)    N)	APIRouterFileFormHTTPException
UploadFile)LogLevel)parse_batch_file)BatchRunService)ensure_known_session)ValidationErrorz/api/workflows/batch.   file
session_id	yaml_filemax_parallel	log_levelc                 4  K   	 t        |d      }|dk  rt        dd      	 | j	                          d {   }t        || j                  xs d      \  }}	d }
|r	 t        |      }
t               }t        j                  |j                  ||||||	|
	             d
||t        |      dS # t        $ r}t        dt        |            d }~ww xY w7 # t        $ r}t        dt        |            d }~ww xY w# t        $ r t        dd      w xY ww)NT)require_connectioni  )status_codedetail   zmax_parallel must be >= 1z	batch.csvz&log_level must be either DEBUG or INFO)r   	file_baser   accepted)statusr   batch_id
task_count)r   r   r   strreadr	   filenamer   
ValueErrorr
   asynciocreate_task	run_batchlen)r   r   r   r   r   managerexccontenttasksr   resolved_levelservices               @/Users/bowang/.openclaw/workspace/ChatDev/server/routes/batch.pyexecute_batchr,      s;    >&zdK a4OPP>		#+GT]]5QkRy N	b%i0N G%$ 	 	

  %j	 A  >CH==> $ >CH==>  	bC8`aa	bss   DB. DC C!C DC> )AD.	C7CCDC 	C; C66C;;D>DD)r!   fastapir   r   r   r   r   entity.enumsr   server.services.batch_parserr	   !server.services.batch_run_servicer
   server.stater   utils.exceptionsr   routerpostr   intr,        r+   <module>r8      s     D D ! 9 = - ,	 #$Cy3i#YQ J.
.. . 	.
 Tz. %.r7   