diff --git a/neuron-explainer/neuron_explainer/explanations/simulator.py b/neuron-explainer/neuron_explainer/explanations/simulator.py index 4111ead..3bb21f0 100644 --- a/neuron-explainer/neuron_explainer/explanations/simulator.py +++ b/neuron-explainer/neuron_explainer/explanations/simulator.py @@ -718,7 +718,7 @@ async def simulate( self.explanation, ) response = await self.api_client.make_request( - prompt=prompt, echo=False, max_tokens=1000 + messages=prompt, max_tokens=1000 ) assert len(response["choices"]) == 1