텔레그램과 랭체인을 연결하여 문서와 링크 축적하기

.env

# 텔레그램 Bot 토큰
TELEGRAM_BOT_TOKEN="7095256272:AAF7eNVPwiJf4do2QShzIRvcTzUXjfu6Wzs"
# 텔레그램 App ID 및 Api hash
TELEGRAM_APP_ID="883836"
TELEGRAM_API_HASH="b94a10597a97ede23136a7e362308702"

telegram_test.py

import os
from dotenv import load_dotenv

from telethon import TelegramClient, events, utils
from telethon.tl import types, functions
from telethon.tl.types import PeerChannel
from telethon.tl.types import Message

load_dotenv()


client = TelegramClient('ohmyelf', os.environ.get("TELEGRAM_APP_ID"), os.environ.get("TELEGRAM_API_HASH"))

# 메세지를 받아오고자 하는 채널들의 채널아이디 목록
channel_ids = {'EvAtZchuno','hantudong','siglab'}

# 내가 메세지를 모으고자 하는 내 봇의 chat id 지정
# chat id를 알아내는 방법은, 아래와 같이 모든 대화목록에서 name 과 id 를 매칭해서 프린트 해보고 찾아내는 방법을 활용했습니다. 더 나은 방법은 차차 알아봐야겠쥬?
# async for dialog in client.iter_dialogs():
#      print(dialog.name, 'has ID', dialog.id)
bot_chat_id = 2197248047

# annotation 을 통해 원하는 특정 채널들의 신규 메세지 이벤트 받아옵니다. 참 쉽쥬?
# chats=channel_ids -> 요걸로 원하는 채널들의 메세지 이벤트만 골라서 받아 수 있습니다.
@client.on(events.NewMessage(chats=channel_ids))
async def my_event_handler(event):
    print("#####")
    print(event.raw_text)
    print("#####")
    print(event.message.forwards)
    print("#####")
    print(event.message.peer_id)
    print("#####")
    print(event.message)

    # 여기에서 들어오는 모든 메세지를 bot_chat_id 채널로 몽땅 포워딩하면 싹 다 모아지겠네요~ 
    # Telethon은, 쌩 API 가지고 환경잡고 테스트 해봤던 것에 비해서는 참 간단하게 잘 만들어진 라이브러리 같아요. 다른 분들은 고민고민하지 마시고 바로 Telethon으로!!! 
    await client.forward_messages(entity=bot_chat_id, messages=event.message, from_peer=event.message.peer_id)

def main():

    client.start()
    client.run_until_disconnected()

main()

import os
from dotenv import load_dotenv

from telethon import TelegramClient, events, utils
from telethon.tl import types, functions
from telethon.tl.types import PeerChannel
from telethon.tl.types import Message

load_dotenv()


# 메시지를 받았을 때, GPT-3 API를 활용해 답변을 작성하는 함수를 정의합니다. 
# llm = ChatOpenAI(model_name="gpt-4o")
llm = ChatOpenAI()
output = StrOutputParser()

#######################################################
# 메세지 요약하기
def generate_response(message):
    
    # 텍스트로부터 URL을 추출한다, 
    urls = re.findall("(?P<url>https?://[^\s]+)", message)

    #   URL 있을 경우 : 해당 url 콘텐츠를 읽어 들여서 요약을 수행하게 하고,
    #   URL 없을 경우 : 메세지를 그대로 읽어 들여서 요약을 수행하게 한다.
    if len(urls) > 0:
        prompt = PromptTemplate.from_template(
            """
            You are my asset manager and analyst with over 20 years of experience. 
            Please read the contents and Write the key points of the contents in 3000 characters or less and if there is links include the links to the bottom.
            if possible, with mark-down and just write in Korean.
            contents : {document}
            links : {links}
            """
            )
        
        url_loader = UnstructuredURLLoader(urls=urls)
        message = url_loader.load()
        chain =  prompt | llm | output
        
        answer = '# 요약 : \n' + chain.invoke({"document":message, "links":urls}) + '\n'
    else :
        prompt = PromptTemplate.from_template(
            """
            You are my asset manager and analyst with over 20 years of experience. 
            Please read the contents and Write the key points of the contents in 3000 characters or less
            if possible, with mark-down and just write in Korean.
            contents : {document}
            """
            )
        chain =  prompt | llm | output
        answer = '# 요약 : \n' + chain.invoke({"document":message}) + '\n'

    return answer
#######################################################

client = TelegramClient('ohmyelf', os.environ.get("TELEGRAM_APP_ID"), os.environ.get("TELEGRAM_API_HASH"))

# 메세지를 받아오고자 하는 채널들의 채널아이디 목록
channel_ids = {'EvAtZchuno','hantudong','siglab'}

# 내가 메세지를 모으고자 하는 내 봇의 chat id 지정
# chat id를 알아내는 방법은, 아래와 같이 모든 대화목록에서 name 과 id 를 매칭해서 프린트 해보고 찾아내는 방법을 활용했습니다. 더 나은 방법은 차차 알아봐야겠쥬?
# async for dialog in client.iter_dialogs():
#      print(dialog.name, 'has ID', dialog.id)
bot_chat_id = 2197248047

# annotation 을 통해 원하는 특정 채널들의 신규 메세지 이벤트 받아옵니다. 참 쉽쥬?
# chats=channel_ids -> 요걸로 원하는 채널들의 메세지 이벤트만 골라서 받아 수 있습니다.
@client.on(events.NewMessage(chats=channel_ids))
async def my_event_handler(event):

    # 여기에서 들어오는 모든 메세지를 bot_chat_id 채널로 몽땅 포워딩하면 싹 다 모아지겠네요~ 
    # Telethon은, 쌩 API 가지고 환경잡고 테스트 해봤던 것에 비해서는 참 간단하게 잘 만들어진 라이브러리 같아요. 다른 분들은 고민고민하지 마시고 바로 Telethon으로!!! 
    # await client.forward_messages(entity=bot_chat_id, messages=event.message, from_peer=event.message.peer_id)

    # 포워딩 대신에 요약결과를 넘겨 보겠습니다.
    answer = generate_response(event.raw_text)
    
    await client.send_message(entity=2197248047, message=answer)
def main():

    client.start()
    client.run_until_disconnected()

main()