|
|
import logging |
|
|
import time |
|
|
import requests |
|
|
import json |
|
|
import os |
|
|
|
|
|
from src.api.finnhub.financial_news_requester import fetch_comp_financial_news |
|
|
from src.api.openrouter.prompt_generator import PromptGenerator |
|
|
|
|
|
|
|
|
class OpenRouterClient: |
|
|
def __init__(self, model: str = "tngtech/deepseek-r1t2-chimera:free", api_key: str | None = None): |
|
|
self._api_key = api_key or os.getenv("OPENROUTER_API_TOKEN") |
|
|
if not self._api_key: |
|
|
raise ValueError("OpenRouter API key is required") |
|
|
self._api_url = "https://openrouter.ai/api/v1/chat/completions" |
|
|
self.model = model |
|
|
|
|
|
def chat(self, user_message: str, role: str = "user") -> str: |
|
|
""" |
|
|
Sends a message to the OpenRouter API and returns the response. |
|
|
Raises RuntimeError for HTTP errors and ValueError for unexpected response formats. |
|
|
|
|
|
Args: |
|
|
user_message (str): The message to send to the model. |
|
|
Returns: |
|
|
str: The response from the model. |
|
|
Raises: |
|
|
RuntimeError: If the API request fails. |
|
|
ValueError: If the response format is unexpected. |
|
|
""" |
|
|
headers = { |
|
|
"Authorization": f"Bearer {self._api_key}", |
|
|
"Content-Type": "application/json" |
|
|
} |
|
|
payload = { |
|
|
"model": self.model, |
|
|
"messages": [ |
|
|
{ |
|
|
"role": role, |
|
|
"content": user_message |
|
|
} |
|
|
] |
|
|
} |
|
|
try: |
|
|
response = requests.post( |
|
|
url=self._api_url, |
|
|
headers=headers, |
|
|
data=json.dumps(payload), |
|
|
timeout=10 |
|
|
) |
|
|
response.raise_for_status() |
|
|
except requests.exceptions.RequestException as e: |
|
|
logging.error(f"OpenRouter request error: {e}") |
|
|
raise RuntimeError(f"Request error: {e}") |
|
|
response_json = response.json() |
|
|
try: |
|
|
return response_json["choices"][0]["message"]["content"] |
|
|
except (KeyError, IndexError): |
|
|
logging.error(f"Unexpected response format: {response_json}") |
|
|
raise ValueError(f"Unexpected response format: {response_json}") |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
sample_news = fetch_comp_financial_news(ticker='NVDA', date_from='2025-08-04', date_to='2025-08-05') |
|
|
print(sample_news) |
|
|
print(len(sample_news)) |
|
|
|
|
|
client = OpenRouterClient() |
|
|
try: |
|
|
for news in sample_news: |
|
|
start_time = time.perf_counter() |
|
|
prompt = PromptGenerator.format_prompt(news) |
|
|
print(f"Prompt: {prompt}") |
|
|
response = client.chat(prompt) |
|
|
print(f"Response: {response}") |
|
|
elapsed = time.perf_counter() - start_time |
|
|
print(f"Processing time: {elapsed:.2f} seconds") |
|
|
except (RuntimeError, ValueError) as e: |
|
|
print(f"Error: {e}") |
|
|
|
