diff --git a/generator/utils/log.py b/generator/utils/log.py index cd8e6dd0..81c6432b 100644 --- a/generator/utils/log.py +++ b/generator/utils/log.py @@ -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: diff --git a/web_api.py b/web_api.py index 6f872951..61be9017 100644 --- a/web_api.py +++ b/web_api.py @@ -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 @@ -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] @@ -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