@@ -44,13 +44,13 @@ def send_request(params: Optional[Union[List, Dict]] = None):
44
44
45
45
46
46
def custom_input (prompt = "" ):
47
- send_request ({"prompt" : prompt })
48
47
try :
48
+ send_request ({"prompt" : prompt })
49
49
headers = get_headers ()
50
50
content_length = int (headers .get ("Content-Length" , 0 ))
51
51
52
52
if content_length :
53
- message_text = STDIN .read (content_length ) # make sure Im getting right content
53
+ message_text = STDIN .read (content_length )
54
54
message_json = json .loads (message_text )
55
55
our_user_input = message_json ["result" ]["userInput" ]
56
56
return our_user_input
@@ -103,7 +103,6 @@ def execute(request, user_globals):
103
103
104
104
105
105
def exec_user_input (user_input , user_globals ):
106
- # have to do redirection
107
106
user_input = user_input [0 ] if isinstance (user_input , list ) else user_input
108
107
109
108
try :
@@ -158,9 +157,8 @@ def get_headers():
158
157
headers = get_headers ()
159
158
content_length = int (headers .get ("Content-Length" , 0 ))
160
159
161
- # TODO: pull out to separate function
162
160
if content_length :
163
- request_text = STDIN .read (content_length ) # make sure Im getting right content
161
+ request_text = STDIN .read (content_length )
164
162
request_json = json .loads (request_text )
165
163
if request_json ["method" ] == "execute" :
166
164
execute (request_json , USER_GLOBALS )
0 commit comments