@@ -45,7 +45,7 @@ def run_record_ide(
4545 knowledge_base_bytes_path = pack_knowledge_base (project_id )
4646
4747 command = [code , record_bytes_path , knowledge_base_bytes_path ]
48- cpu_limit = docker .types .Ulimit (name = "cpu" , soft = 25 , hard = 25 )
48+ cpu_limit = docker .types .Ulimit (name = "cpu" , soft = 50 , hard = 50 )
4949 container_name = str (uuid .uuid4 ())
5050 container = client .containers .create (
5151 command = command ,
@@ -61,10 +61,14 @@ def run_record_ide(
6161 knowledge_base_tar_path = f"{ project_id } .tar"
6262
6363 copy_to (
64- f"./{ record_bytes_path } " , f"{ container .name } :/{ record_bytes_path } " , record_tar_path
64+ f"./{ record_bytes_path } " ,
65+ f"{ container .name } :/{ record_bytes_path } " ,
66+ record_tar_path ,
6567 )
6668 copy_to (
67- f"./{ knowledge_base_bytes_path } " , f"{ container .name } :/{ knowledge_base_bytes_path } " , knowledge_base_tar_path
69+ f"./{ knowledge_base_bytes_path } " ,
70+ f"{ container .name } :/{ knowledge_base_bytes_path } " ,
71+ knowledge_base_tar_path ,
6872 )
6973 daemon .run (cancel_container , container_name , container )
7074 __containers_running [container_name ] = True
@@ -99,7 +103,7 @@ def run_record_ide(
99103
100104
101105def cancel_container (name : str , container : Any ):
102- TIMEOUT = 15
106+ TIMEOUT = 60
103107 time .sleep (TIMEOUT )
104108 if name in __containers_running and __containers_running [name ]:
105109 __containers_running [name ] = False
@@ -133,9 +137,10 @@ def pack_record_data(project_id: str, record_id: str) -> str:
133137 pickle .dump (full_data , file )
134138 return record_bytes_path
135139
140+
136141def pack_knowledge_base (project_id : str ) -> str :
137142 knowledge_base_source = knowledge_base .build_knowledge_base_from_project (project_id )
138143 knowledge_base_path = f"{ project_id } knowledge_base.p"
139144 with open (knowledge_base_path , "wb" ) as file :
140145 pickle .dump (knowledge_base_source , file )
141- return knowledge_base_path
146+ return knowledge_base_path
0 commit comments