| 58 | def __init__(self, model: str) -> None: |
| 59 | super().__init__(model) |
| 60 | def wrap_query_results(self, query_results: List[Dict]) -> str: |
| 61 | wrapped_query_results = "" |
| 62 | for result in query_results: |
| 63 | wrapped_query_results += f"File: {result['file']}\n" |
| 64 | wrapped_query_results += f"Content: {result['content'][:300]}...\n" |
| 65 | wrapped_query_results += "---" |
| 66 | return wrapped_query_results |
| 67 | def wrap_reranked_results(self, reranked_paths: List[str]) -> str: |
| 68 | wrapped_reranked_results = "[Referenced code files]:" |
| 69 | for path in reranked_paths: |