amirkiarafiei commited on
Commit
5370a22
·
1 Parent(s): df0da5b

refactor: enhance logging with shorter loggings

Browse files
Files changed (1) hide show
  1. main.py +8 -5
main.py CHANGED
@@ -98,7 +98,8 @@ async def multi_module_bug_localization(request: MultiModuleRequest, ctx: Contex
98
  )
99
  response.raise_for_status()
100
 
101
- logger.info(f"Received response: {response.text}")
 
102
 
103
  return {"issue_key": request.issue_key, "result": response.json()}
104
 
@@ -159,7 +160,7 @@ async def single_module_bug_localization(request: SingleModuleRequest, ctx: Cont
159
  # Generate a UUID for issue_key even if provided (overwrite)
160
  request.issue_key = str(uuid.uuid4())
161
 
162
- logger.info(f"Sending single-module bug localization request to {api_url} with data: {request.model_dump()}")
163
 
164
  response = await client.post(
165
  api_url,
@@ -175,7 +176,8 @@ async def single_module_bug_localization(request: SingleModuleRequest, ctx: Cont
175
  )
176
  response.raise_for_status()
177
 
178
- logger.info(f"Received response: {response.text}")
 
179
 
180
  return {"issue_key": request.issue_key, "result": response.json()}
181
 
@@ -234,7 +236,7 @@ async def search_space_routing(request: SearchSpaceRoutingRequest, ctx: Context)
234
  # Generate a UUID for issue_key even if provided (overwrite)
235
  request.issue_key = str(uuid.uuid4())
236
 
237
- logger.info(f"Sending search space routing request to {api_url} with data: {request.model_dump()}")
238
 
239
  response = await client.post(
240
  api_url,
@@ -249,7 +251,8 @@ async def search_space_routing(request: SearchSpaceRoutingRequest, ctx: Context)
249
  )
250
  response.raise_for_status()
251
 
252
- logger.info(f"Received response: {response.text}")
 
253
 
254
  return {"issue_key": request.issue_key, "result": response.json()}
255
 
 
98
  )
99
  response.raise_for_status()
100
 
101
+ logger.info(f"Received response: {str(response.text)[:150]}...")
102
+ logger.debug(f"Response JSON: {response.json()}")
103
 
104
  return {"issue_key": request.issue_key, "result": response.json()}
105
 
 
160
  # Generate a UUID for issue_key even if provided (overwrite)
161
  request.issue_key = str(uuid.uuid4())
162
 
163
+ logger.info(f"Sending single-module bug localization request to {api_url} with data: {str(request.model_dump())[:50]}...")
164
 
165
  response = await client.post(
166
  api_url,
 
176
  )
177
  response.raise_for_status()
178
 
179
+ logger.info(f"Received response: {str(response.text)[:150]}...")
180
+ logger.debug(f"Response JSON: {response.json()}")
181
 
182
  return {"issue_key": request.issue_key, "result": response.json()}
183
 
 
236
  # Generate a UUID for issue_key even if provided (overwrite)
237
  request.issue_key = str(uuid.uuid4())
238
 
239
+ logger.info(f"Sending search space routing request to {api_url} with data: {str(request.model_dump())[:50]}...")
240
 
241
  response = await client.post(
242
  api_url,
 
251
  )
252
  response.raise_for_status()
253
 
254
+ logger.info(f"Received response: {str(response.text)[:150]}...")
255
+ logger.debug(f"Response JSON: {response.json()}")
256
 
257
  return {"issue_key": request.issue_key, "result": response.json()}
258