Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

RAG機能のPoCの実装 #28

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
Open

RAG機能のPoCの実装 #28

wants to merge 6 commits into from

Conversation

kaitoyama
Copy link
Contributor

@kaitoyama kaitoyama commented Oct 24, 2024

User description

wikiのデータの一部をembeddingした結果を使ったvectorDB形式のRAGのプロトタイプ


PR Type

enhancement, dependencies


Description

  • 新しいragパッケージを追加し、RAG機能を実装しました。これにより、OpenAIのチャットストリームとMilvusを使用した検索が可能になりました。
  • ステージング機能にリセットと画像取得のロジックを追加し、rag.Chat関数を呼び出すようにしました。
  • go.modgo.sumに新しい依存関係を追加し、ライブラリのバージョン情報を更新しました。

Changes walkthrough 📝

Relevant files
Enhancement
MessageReceived.go
ステージング機能にRAGチャットロジックを追加                                                                   

internal/handler/MessageReceived.go

  • ragパッケージのインポートを追加
  • ステージング機能にリセットと画像取得のロジックを追加
  • rag.Chat関数を呼び出すロジックを追加
  • +10/-0   
    rag.go
    RAG機能の実装とチャットメッセージ管理の追加                                                                   

    internal/rag/rag.go

  • 新しいragパッケージを追加
  • OpenAIのチャットストリームとMilvusを使用した検索ロジックを実装
  • チャットメッセージの管理とシステムメッセージの操作を実装
  • +370/-0 
    Dependencies
    go.mod
    新しい依存関係の追加と更新                                                                                       

    go.mod

    • 新しい依存関係を追加
    • milvus-sdk-gogo-openaiなどのライブラリを追加
    +22/-1   
    go.sum
    依存関係のチェックサムとバージョン情報の更新                                                                     

    go.sum

    • 依存関係のチェックサムを追加
    • 新しいライブラリのバージョン情報を追加
    +367/-0 

    💡 PR-Agent usage: Comment /help "your question" on any pull request to receive relevant information

    return
    }

    func Chat(channelID, newMessageText string, imageBase64 []string) {
    Copy link
    Contributor Author

    Choose a reason for hiding this comment

    The reason will be displayed to describe this comment to others. Learn more.

    ここ以外はgptディレクトリからのコピペです

    @kaitoyama kaitoyama requested a review from pikachu0310 October 24, 2024 17:05
    @kaitoyama kaitoyama self-assigned this Oct 24, 2024
    @kaitoyama kaitoyama marked this pull request as ready for review October 24, 2024 17:05
    Copy link

    PR Reviewer Guide 🔍

    Here are some key observations to aid the review process:

    ⏱️ Estimated effort to review: 4 🔵🔵🔵🔵⚪
    🧪 No relevant tests
    🔒 No security concerns identified
    ⚡ Recommended focus areas for review

    Error Handling
    エラーハンドリングが不十分で、いくつかの関数でエラーが発生した場合に適切な処理が行われていない可能性があります。特に、OpenAIStream 関数内でのエラーハンドリングを見直す必要があります。

    Logging
    log.Fatal を使用している箇所が複数存在します。これにより、アプリケーションが予期せず終了する可能性があります。エラーログを記録後、適切に処理を続行するか、ユーザーに通知する方法を検討してください。

    Hardcoded Values
    ハードコードされた値(例: entity.NewIndexHNSWSearchParam(74))が使用されています。これらの値を設定ファイルや環境変数から取得するように変更することで、柔軟性とメンテナンスの容易さを向上させることができます。

    @kaitoyama kaitoyama requested a review from nokhnaton November 10, 2024 07:26
    Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
    Projects
    None yet
    Development

    Successfully merging this pull request may close these issues.

    1 participant