feat: add sentiment analyst agent
This commit is contained in:
@@ -0,0 +1,38 @@
|
||||
import { describe, it, expect } from "vitest";
|
||||
import { SentimentAnalyst } from "../sentiment";
|
||||
import type { OpenRouterClient } from "../../lib/openrouter";
|
||||
|
||||
describe("SentimentAnalyst", () => {
|
||||
it("should analyze sentiment from headlines", async () => {
|
||||
const mockClient = {
|
||||
createChatCompletion: async () => ({
|
||||
choices: [
|
||||
{
|
||||
message: {
|
||||
content:
|
||||
'{"signal":"bullish","confidence":0.85,"reasoning":"Positive sentiment from news"}',
|
||||
},
|
||||
},
|
||||
],
|
||||
}),
|
||||
} as unknown as OpenRouterClient;
|
||||
|
||||
const analyst = new SentimentAnalyst(mockClient);
|
||||
const result = await analyst.analyze("AAPL", {
|
||||
headlines: ["Apple beats earnings", "New iPhone launch"],
|
||||
source: "news",
|
||||
});
|
||||
|
||||
expect(result.analyst).toBe("sentiment");
|
||||
expect(result.signal.signal).toBe("bullish");
|
||||
});
|
||||
|
||||
it("should use specified model", () => {
|
||||
const mockClient = {} as unknown as OpenRouterClient;
|
||||
const analyst = new SentimentAnalyst(mockClient, {
|
||||
model: "custom/model",
|
||||
});
|
||||
|
||||
expect(analyst.getModel()).toBe("custom/model");
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user