File size: 928 Bytes
f1d5e1c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import asyncio
import os
from dotenv import load_dotenv

load_dotenv()
import sys

sys.path.append(".")

async def test_deep_research():
    from src.utils.deep_research import deep_research
    from src.utils import utils
    
    task = "write a report about DeepSeek-R1, get its pdf"
    llm = utils.get_llm_model(
        provider="gemini",
        model_name="gemini-2.0-flash-thinking-exp-01-21",
        temperature=1.0,
        api_key=os.getenv("GOOGLE_API_KEY", "")
    )
    
    report_content, report_file_path = await deep_research(task=task, llm=llm, agent_state=None, 
                                                           max_search_iterations=1, 
                                                           max_query_num=3,
                                                           use_own_browser=False)
    


if __name__ == "__main__":
    asyncio.run(test_deep_research())