diff --git a/llm_consortium/__init__.py b/llm_consortium/__init__.py index 8201189..6053951 100644 --- a/llm_consortium/__init__.py +++ b/llm_consortium/__init__.py @@ -224,9 +224,7 @@ async def _get_model_response(self, model: str, prompt: str) -> Dict[str, Any]: {prompt} """ response = llm.get_model(model).prompt(xml_prompt, system=self.system_prompt) - - # Await the text from the response. - text = await response.text() + text = response.text() asyncio.create_task(log_response(response, model)) return { @@ -336,16 +334,16 @@ async def _synthesize_responses(self, original_prompt: str, responses: List[Dict # Print raw arbiter response click.echo("\nArbiter Response:\n") - click.echo(await arbiter_response.text()) + click.echo(arbiter_response.text()) click.echo("\n---\n") try: - return self._parse_arbiter_response(await arbiter_response.text()) + return self._parse_arbiter_response(arbiter_response.text()) except Exception as e: logger.error(f"Error parsing arbiter response: {e}") return { - "synthesis": await arbiter_response.text(), + "synthesis": arbiter_response.text(), "confidence": 0.5, "analysis": "Parsing failed - see raw response", "dissent": "",