Fix Pydantic compatibility and improve error handling

OpenAITools/CrinicalTrialTools.py

@@ -24,8 +24,8 @@ except ImportError:
 from langchain_core.runnables import RunnablePassthrough
 from langchain_core.output_parsers import StrOutputParser
 
-# GPT初期化
-gpt = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
+# 修正: モジュールレベルでの初期化を削除
+# gpt = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
 
 ## Database関連のインポート
 from langchain_community.utilities import SQLDatabase
@@ -36,6 +36,15 @@ from langchain_core.prompts import MessagesPlaceholder
 
 from OpenAITools.FetchTools import fetch_clinical_trials, fetch_clinical_trials_jp
 
+# GPT初期化を関数として追加
+def get_openai_client():
+    """OpenAIクライアントを安全に初期化"""
+    try:
+        return ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
+    except Exception as e:
+        print(f"OpenAI初期化エラー: {e}")
+        return None
+
 # エラーハンドリング用のデコレータ
 def retry_on_error(max_retries=3, delay=1):
     """エラー時のリトライデコレータ"""

app.py

@@ -2,20 +2,54 @@ import gradio as gr
 import pandas as pd
 import time
 import traceback
+import os
 from OpenAITools.FetchTools import fetch_clinical_trials
 from langchain_openai import ChatOpenAI
 from langchain_groq import ChatGroq
 from OpenAITools.CrinicalTrialTools import SimpleClinicalTrialAgent, GraderAgent, LLMTranslator, generate_ex_question_English
 
-# モデルとエージェントの初期化
-groq = ChatGroq(model_name="llama3-70b-8192", temperature=0)
-translator = LLMTranslator(groq)
-CriteriaCheckAgent = SimpleClinicalTrialAgent(groq)
-grader_agent = GraderAgent(groq)
+# 環境変数チェック
+def check_environment():
+    """環境変数をチェックし、不足している場合は警告"""
+    missing_vars = []
+    
+    if not os.getenv("GROQ_API_KEY"):
+        missing_vars.append("GROQ_API_KEY")
+    
+    if not os.getenv("OPENAI_API_KEY"):
+        missing_vars.append("OPENAI_API_KEY")
+    
+    if missing_vars:
+        print(f"⚠️ 環境変数が設定されていません: {', '.join(missing_vars)}")
+        print("一部の機能が制限される可能性があります。")
+    
+    return len(missing_vars) == 0
+
+# 環境変数チェック実行
+env_ok = check_environment()
+
+# モデルとエージェントの安全な初期化
+def safe_init_agents():
+    """エージェントを安全に初期化"""
+    try:
+        groq = ChatGroq(model_name="llama3-70b-8192", temperature=0)
+        translator = LLMTranslator(groq)
+        criteria_agent = SimpleClinicalTrialAgent(groq)
+        grader_agent = GraderAgent(groq)
+        return translator, criteria_agent, grader_agent
+    except Exception as e:
+        print(f"エージェント初期化エラー: {e}")
+        return None, None, None
+
+# エージェント初期化
+translator, CriteriaCheckAgent, grader_agent = safe_init_agents()
 
 # エラーハンドリング付きでエージェント評価を実行する関数
 def evaluate_with_retry(agent, criteria, question, max_retries=3):
     """エラーハンドリング付きでエージェント評価を実行"""
+    if agent is None:
+        return "評価エラー: エージェントが初期化されていません。API keyを確認してください。"
+    
     for attempt in range(max_retries):
         try:
             return agent.evaluate_eligibility(criteria, question)
@@ -32,6 +66,8 @@ def evaluate_with_retry(agent, criteria, question, max_retries=3):
                     continue
                 else:
                     return "評価エラー: サーバーに接続できませんでした"
+            elif "API key" in str(e) or "authentication" in str(e).lower():
+                return "評価エラー: API keyが無効または設定されていません"
             else:
                 print(f"予期しないエラー (試行 {attempt + 1}/{max_retries}): {e}")
                 if attempt < max_retries - 1:
@@ -43,6 +79,9 @@ def evaluate_with_retry(agent, criteria, question, max_retries=3):
 
 def evaluate_grade_with_retry(agent, judgment, max_retries=3):
     """エラーハンドリング付きでグレード評価を実行"""
+    if agent is None:
+        return "unclear"
+    
     for attempt in range(max_retries):
         try:
             return agent.evaluate_eligibility(judgment)
@@ -54,6 +93,8 @@ def evaluate_grade_with_retry(agent, judgment, max_retries=3):
                     continue
                 else:
                     return "unclear"
+            elif "API key" in str(e) or "authentication" in str(e).lower():
+                return "unclear"
             else:
                 print(f"予期しないエラー (グレード評価 - 試行 {attempt + 1}/{max_retries}): {e}")
                 if attempt < max_retries - 1:
@@ -72,7 +113,11 @@ def generate_dataframe(age, sex, tumor_type, GeneMutation, Meseable, Biopsiable)
         
         # 日本語の腫瘍タイプを英語に翻訳
         try:
-            TumorName = translator.translate(tumor_type)
+            if translator is not None:
+                TumorName = translator.translate(tumor_type)
+            else:
+                print("翻訳エージェントが利用できません。元の値を使用します。")
+                TumorName = tumor_type
         except Exception as e:
             print(f"翻訳エラー: {e}")
             TumorName = tumor_type  # 翻訳に失敗した場合は元の値を使用
@@ -162,7 +207,14 @@ def download_full_csv(df):
 # Gradioインターフェースの作成
 with gr.Blocks(title="臨床試験適格性評価", theme=gr.themes.Soft()) as demo:
     gr.Markdown("## 臨床試験適格性評価インターフェース")
-    gr.Markdown("⚠️ **注意**: サーバーエラーが発生する場合があります。エラーが続く場合は少し時間をおいてから再試行してください。")
+    
+    # 環境変数状態の表示
+    if env_ok:
+        gr.Markdown("✅ **ステータス**: 全ての環境変数が設定されています")
+    else:
+        gr.Markdown("⚠️ **���意**: 一部の環境変数が設定されていません。機能が制限される可能性があります。")
+    
+    gr.Markdown("💡 **使用方法**: 患者情報を入力して「Generate Clinical Trials Data」をクリックしてください。")
 
     # 各種入力フィールド
     with gr.Row():

requirements.txt

@@ -6,23 +6,23 @@ pandas>=1.5.0
 numpy>=1.21.0
 requests>=2.25.0
 
-# LangChain ecosystem (安定バージョンに固定)
-langchain==0.3.2
-langchain-community==0.3.1
-langchain-core==0.3.9
-langchain-openai==0.2.2
-langchain-groq==0.2.0
-langchain-text-splitters==0.3.0
+# LangChain ecosystem (互換性のあるバージョンに調整)
+langchain==0.2.16
+langchain-community==0.2.16
+langchain-core==0.2.38
+langchain-openai==0.1.23
+langchain-groq==0.1.9
+langchain-text-splitters==0.2.4
 
 # LLM providers
-openai>=1.0.0
+openai>=1.0.0,<2.0.0
 groq>=0.4.0
 
 # Database utilities
 SQLAlchemy>=2.0.0,<3.0.0
 
-# Pydantic (Gradio互換性のため)
-pydantic>=2.0.0,<3.0.0
+# Pydantic (LangChain互換性のため)
+pydantic>=2.5.0,<3.0.0
 
 # Text processing (軽量化)
 tiktoken>=0.4.0
@@ -31,11 +31,6 @@ tiktoken>=0.4.0
 tenacity>=8.0.0
 packaging>=23.0.0
 
-# Optional: Vector databases (必要な場合のみ)
-# chromadb>=0.4.0
-# sentence-transformers>=2.2.0
-# transformers>=4.40.0
-
 # Optional: Bio-related libraries (if needed for your specific use case)
 # Uncomment the following lines if bio-processing is required
 # biopython>=1.83