반응형

개인 프로젝트는 작년 이맘 때 개발을 끝냈다.

오랜만에 블로그에 들어왔는데 개발 결과물이 없어 마음이 불편하여 기록하려고 한다.

실제로 내가 제일 사용을 많이 했었고, 지금도 가끔 사용하고 있는  InvestorGPT에 대해서 기록한다.

 

Agent

agent = initialize_agent(
    llm=llm,
    verbose=True,
    agent=AgentType.OPENAI_FUNCTIONS,
    handle_parsing_errors=True,
    tools=[
        CompanyIncomeStatementTool(),
        CompanyStockPerformanceTool(),
        # StockMarketSymbolSearchTool(),
        CompanyOverviewTool(),
    ],
    agent_kwargs={
        "system_message": SystemMessage(
            content="""
            You are a hedge fund manager.
            
            You evaluate a company and provide your opinion and reasons why the stock is a buy or not.
            
            Consider the performance of a stock, the company overview and the income statement.
            
            Be assertive in your judgement and recommend the stock or advise the user against it.
        """
        )
    },
)

 

StockMargetSymbolSearchTool은 회사 이름을 입력하면 티커를 찾아주는 툴이다.

개발 당시 DuckDuckGo가 동작을 잘 했었는데 업데이트 하면서 동작을 하지 않길래 제외하였다.

내 성격상 GPT에 물어볼 정도면 티커는 알고 있다.

 

 

개발 결과물

streamlit으로 간단하게 웹 ui를 개발 했고 혼자 사용하기 때문에 local server에서 실행시킨다.

왼쪽 목록에 DocumentGPT는 문서를 올리면 embedding -> vectorDB -> RAG 적용 후 내용을 검색하는 GPT이다.

PrivateGPT는 local LLM(Ollma)을 사용하는 GPT이고,

QuizGPT는 파일을 올리거나 위키피디아에서 검색을 하면 문제를 생성해주는 GPT이다.

PrivateGPT와 QuizGPT에도 DocumentGPT와 동일하게 문서를 embedding -> vectorDB -> RAG 가 적용되어 있다.

 

InvestorGPT - 개발 결과물

 

비용

개발 당시 최신 모델은 비싸서 gpt-3.5-turbo-1106을 사용했었다. 아래 0.02 달러는 위 스샷 내용을 한번 질의한 비용이다.

최근에 모델들이 많이 업데이트 됐던데 바꿔가면서 테스트 해봐야 할 것 같다.

1회 질문 Token 비용

 

반응형

+ Recent posts