Skip to content

Commit

Permalink
log more endpoints
Browse files Browse the repository at this point in the history
  • Loading branch information
djstrong committed Nov 30, 2023
1 parent ff77123 commit a5b8551
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 0 deletions.
1 change: 1 addition & 0 deletions generator/utils/log.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ def create_entry(self, request: dict) -> dict:
def create_grouped_log_entry(self, request: dict, result: dict[str, list[GeneratedName]]) -> dict:
entry = self.create_entry(request)
entry['response'] = {str(category): [self.convert_suggestion_to_log_entry(gn) for gn in gns] for category, gns in result.items()} #FIXME better logging system
entry['endpoint'] = 'suggestions_by_category'
return entry

def create_log_entry(self, request: dict, result: list[GeneratedName]) -> dict:
Expand Down
6 changes: 6 additions & 0 deletions web_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -336,6 +336,8 @@ async def sample_collection_members(sample_command: SampleCollectionMembers):

response = convert_to_suggestion_format(sampled_members, include_metadata=sample_command.metadata)

logger.info(json.dumps({'endpoint': 'sample_collection_members', 'request': sample_command.model_dump()}))

return response


Expand Down Expand Up @@ -368,6 +370,8 @@ async def fetch_top_collection_members(fetch_top10_command: Top10CollectionMembe
response2 = convert_related_to_grouped_suggestions_format({result['collection_title']: rs},
include_metadata=fetch_top10_command.metadata)

logger.info(json.dumps({'endpoint': 'fetch_top_collection_members', 'request': fetch_top10_command.model_dump()}))

return response2[0]


Expand All @@ -391,6 +395,8 @@ async def scramble_collection_tokens(scramble_command: ScrambleCollectionTokens)

response = convert_to_suggestion_format(suggestions, include_metadata=scramble_command.metadata)

logger.info(json.dumps({'endpoint': 'scramble_collection_tokens', 'request': scramble_command.model_dump()}))

return response


Expand Down

0 comments on commit a5b8551

Please sign in to comment.