18 lines
434 B
Plaintext
18 lines
434 B
Plaintext
|
|
if self.test_mode:
|
|
print(prompt)
|
|
return "[TEST MODE]"
|
|
|
|
if not self.client:
|
|
raise ValueError("LLM client not initialized. Please provide a valid API key.")
|
|
|
|
message = self.client.messages.create(
|
|
model=self.model,
|
|
max_tokens=2048,
|
|
messages=[{"role": "user", "content": prompt}],
|
|
)
|
|
|
|
return message.content[0].text
|
|
|
|
|