DeL-TaiseiOzaki commited on
Commit
2abab5f
·
1 Parent(s): 230b1a5
Files changed (3) hide show
  1. README.md +1 -1
  2. config/llm_setting.py +11 -16
  3. requirements.txt +1 -0
README.md CHANGED
@@ -8,7 +8,7 @@ sdk_version: 1.39.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
11
- short_description: プログラミング関連ファイルを再帰的にスキャンし、内容を単一のテキストファイルにエクスポートするツールです。GitHubリ
12
  ---
13
 
14
  # get_repository_info_by_llm
 
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
11
+ short_description: プログラミング関連ファイルを再帰的にスキャンし、内容を単一のテキストファイルにエクスポートするツールです。
12
  ---
13
 
14
  # get_repository_info_by_llm
config/llm_setting.py CHANGED
@@ -1,24 +1,19 @@
1
  import os
2
  from dotenv import load_dotenv
3
- from typing import Literal
4
 
5
  class LLMSettings:
6
  def __init__(self):
7
  load_dotenv()
8
-
9
- self.openai_api_key = os.getenv('OPENAI_API_KEY')
10
- self.anthropic_api_key = os.getenv('ANTHROPIC_API_KEY')
11
- self.default_llm = os.getenv('DEFAULT_LLM', 'claude')
12
-
13
- # API キーの存在確認
14
- if not self.openai_api_key and not self.anthropic_api_key:
15
- raise ValueError("少なくとも1つのAPIキーが必要です。")
16
-
17
- def get_available_models(self) -> list[Literal['claude', 'gpt']]:
18
- """利用可能なモデルのリストを返す"""
19
- models = []
20
  if self.anthropic_api_key:
21
- models.append('claude')
22
  if self.openai_api_key:
23
- models.append('gpt')
24
- return models
 
 
 
1
  import os
2
  from dotenv import load_dotenv
 
3
 
4
  class LLMSettings:
5
  def __init__(self):
6
  load_dotenv()
7
+ self.anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
8
+ self.openai_api_key = os.getenv("OPENAI_API_KEY")
9
+ self.default_llm = "claude"
10
+
11
+ def get_available_models(self):
12
+ available_models = []
 
 
 
 
 
 
13
  if self.anthropic_api_key:
14
+ available_models.append("claude")
15
  if self.openai_api_key:
16
+ available_models.append("openai")
17
+ if not available_models:
18
+ raise ValueError("APIキーが設定されていません")
19
+ return available_models
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
  streamlit
2
  openai
 
3
  gitpython
4
  python-dotenv
5
  pathlib
 
1
  streamlit
2
  openai
3
+ anthropic
4
  gitpython
5
  python-dotenv
6
  pathlib