tests: testing modes have been added in an attempt to tune without wasting tokens.

This commit is contained in:
2026-02-19 22:46:15 -05:00
parent b8566fc2af
commit 6882e53280
6 changed files with 447 additions and 17 deletions
+17
View File
@@ -0,0 +1,17 @@
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