(documents_df)
| 110 | |
| 111 | |
| 112 | def process_retrieval_ducoment(documents_df): |
| 113 | ir_corpus = {} |
| 114 | corpus2tool = {} |
| 115 | for row in documents_df.itertuples(): |
| 116 | doc = json.loads(row.document_content) |
| 117 | ir_corpus[row.docid] = (doc.get('category_name', '') or '') + ', ' + \ |
| 118 | (doc.get('tool_name', '') or '') + ', ' + \ |
| 119 | (doc.get('api_name', '') or '') + ', ' + \ |
| 120 | (doc.get('api_description', '') or '') + \ |
| 121 | ', required_params: ' + json.dumps(doc.get('required_parameters', '')) + \ |
| 122 | ', optional_params: ' + json.dumps(doc.get('optional_parameters', '')) + \ |
| 123 | ', return_schema: ' + json.dumps(doc.get('template_response', '')) |
| 124 | corpus2tool[(doc.get('category_name', '') or '') + ', ' + \ |
| 125 | (doc.get('tool_name', '') or '') + ', ' + \ |
| 126 | (doc.get('api_name', '') or '') + ', ' + \ |
| 127 | (doc.get('api_description', '') or '') + \ |
| 128 | ', required_params: ' + json.dumps(doc.get('required_parameters', '')) + \ |
| 129 | ', optional_params: ' + json.dumps(doc.get('optional_parameters', '')) + \ |
| 130 | ', return_schema: ' + json.dumps(doc.get('template_response', ''))] = doc['category_name'] + '\t' + doc['tool_name'] + '\t' + doc['api_name'] |
| 131 | return ir_corpus, corpus2tool |
| 132 |
no outgoing calls
no test coverage detected