Method predict

utils/claude_inference.py:13–23

(self, system_prompt, prompt)

        )

    def predict(self, system_prompt, prompt):
        message = self.client.messages.create(
            model="claude-3-5-sonnet-20240620",
            max_tokens=1024,
            system=system_prompt,
            messages=[
                {"role": "user", "content": prompt}
            ],
            temperature=0,
        )
        return self.post_process(message.content[0].text)

    def post_process(self, text):
        match = re.search(r"(?i)(?<=\banswer:\s).*", text)