517842c834
* Aggiorna gli agenti e il modello del team per utilizzare OLLAMA_QWEN_1B * Riorganizza e rinomina funzioni di estrazione in moduli di mercato e notizie; migliora la gestione delle importazioni * Spostato main nel corretto file __main__ e aggiornato il README.md * Aggiunta cartella per i modelli, agenti e team * Aggiornata la posizione delle istruzioni * Rimossi TODO e Aggiunto documentazione per metodi aggregated * Aggiornate le istruzioni del coordinatore del team * utils type checks * Rinominato BaseWrapper in MarketWrapper e fix type check markets * fix type checks di notizie e social. * Aggiunti type hints finali * Riorganizzati gli import * Refactoring architetturale e spostamento classi base - Eliminazione del file __init__.py obsoleto che importava ChatManager e Pipeline - Spostamento della classe Pipeline in agents/pipeline.py - Spostamento della classe ChatManager in utils/chat_manager.py - Aggiornamento di __main__.py per importare da app.utils e app.agents, e modifica della logica per utilizzare Pipeline invece di chat per la selezione di provider e stile - Creazione della cartella base con classi base comuni: markets.py (ProductInfo, Price, MarketWrapper), news.py (Article, NewsWrapper), social.py (SocialPost, SocialComment, SocialWrapper) - Aggiornamento di tutti gli import nel progetto (markets/, news/, social/, utils/, tests/) per utilizzare la nuova struttura base/ * Aggiornato Readme * Corretto il valore predefinito della valuta in BinanceWrapper da "USDT" a "USD" * fix type in tests * fix type per models * Rinominato 'quote_currency' in 'currency' e aggiornato il trattamento del timestamp in Price * fix errors found by Copilot * WrapperHandler: semplificata la logica di chiamata delle funzioni sui wrapper * fix docs * fix demos, semplificata logica lista ollama
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
import pytest
|
|
from app.news import NewsAPIsTool
|
|
|
|
|
|
@pytest.mark.tools
|
|
@pytest.mark.news
|
|
@pytest.mark.api
|
|
class TestNewsAPITool:
|
|
def test_news_api_tool(self):
|
|
tool = NewsAPIsTool()
|
|
assert tool is not None
|
|
|
|
def test_news_api_tool_get_top(self):
|
|
tool = NewsAPIsTool()
|
|
result = tool.wrapper_handler.try_call(lambda w: w.get_top_headlines(limit=2))
|
|
assert isinstance(result, list)
|
|
assert len(result) > 0
|
|
for article in result:
|
|
assert article.title is not None
|
|
assert article.source is not None
|
|
|
|
def test_news_api_tool_get_latest(self):
|
|
tool = NewsAPIsTool()
|
|
result = tool.wrapper_handler.try_call(lambda w: w.get_latest_news(query="crypto", limit=2))
|
|
assert isinstance(result, list)
|
|
assert len(result) > 0
|
|
for article in result:
|
|
assert article.title is not None
|
|
assert article.source is not None
|
|
|
|
def test_news_api_tool_get_top__all_results(self):
|
|
tool = NewsAPIsTool()
|
|
result = tool.wrapper_handler.try_call_all(lambda w: w.get_top_headlines(limit=2))
|
|
assert isinstance(result, dict)
|
|
assert len(result.keys()) > 0
|
|
for _provider, articles in result.items():
|
|
for article in articles:
|
|
assert article.title is not None
|
|
assert article.source is not None
|
|
|
|
def test_news_api_tool_get_latest__all_results(self):
|
|
tool = NewsAPIsTool()
|
|
result = tool.wrapper_handler.try_call_all(lambda w: w.get_latest_news(query="crypto", limit=2))
|
|
assert isinstance(result, dict)
|
|
assert len(result.keys()) > 0
|
|
for _provider, articles in result.items():
|
|
for article in articles:
|
|
assert article.title is not None
|
|
assert article.source is not None
|