deploy Nous v16 update
This commit is contained in:
@@ -137,6 +137,22 @@ class BrowserOnlyAppTests(unittest.TestCase):
|
||||
server.http_json("GET", "https://example.test/v1/models", error_prefix="ai")
|
||||
self.assertEqual(open_url.call_args.args[0].headers.get("User-agent"), server.OUTBOUND_USER_AGENT)
|
||||
|
||||
def test_ai_assessment_is_saved_for_a_prospect(self):
|
||||
conn = server.db()
|
||||
conn.execute("INSERT INTO prospects(name,domain,category,website_status,opportunity_score,discovered_at) VALUES(?,?,?,?,?,?)", ("Acme Plumbing", "acme.co.za", "plumber", "no_detected_website", 82, server.now()))
|
||||
conn.execute("INSERT INTO settings(key,value,updated_at) VALUES(?,?,?)", ("AI_PROVIDER", "nous", server.now()))
|
||||
conn.execute("INSERT INTO settings(key,value,updated_at) VALUES(?,?,?)", ("AI_MODEL", "meituan/longcat-2.0:free", server.now()))
|
||||
conn.execute("INSERT INTO settings(key,value,updated_at) VALUES(?,?,?)", ("AI_API_KEY", "test-key", server.now()))
|
||||
conn.commit(); conn.close()
|
||||
response = {"choices": [{"message": {"content": '{"probability":84,"recommendation":"contact","reasoning":"Active business with a clear website opportunity."}'}}]}
|
||||
with patch("api.server.http_json", return_value=response):
|
||||
status, body = self.request("POST", "/api/prospects/1/assess", {})
|
||||
self.assertEqual(status, 200)
|
||||
self.assertEqual(body["ai_probability"], 84)
|
||||
self.assertEqual(body["ai_recommendation"], "contact")
|
||||
saved = server.db().execute("SELECT ai_probability,ai_recommendation,ai_reasoning FROM prospects WHERE id=1").fetchone()
|
||||
self.assertEqual(tuple(saved), (84.0, "contact", "Active business with a clear website opportunity."))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user