DeL-TaiseiOzaki commited on
Commit
443a8ae
·
1 Parent(s): 9477a09
Files changed (1) hide show
  1. services/llm_service.py +8 -3
services/llm_service.py CHANGED
@@ -48,16 +48,21 @@ class LLMService:
48
  self.conversation_history = []
49
 
50
  def get_response(self, content: str, query: str) -> tuple[Optional[str], Optional[str]]:
 
51
  try:
52
  prompt = self.create_prompt(content, query)
53
  self._add_to_history("user", prompt)
54
 
 
55
  response = self.claude_client.messages.create(
56
- model="claude-3-5-sonnet-latest",
57
- messages=self._format_messages_for_claude()
 
 
 
58
  )
59
- answer = response.content[0].text
60
 
 
61
  self._add_to_history("assistant", answer)
62
  return answer, None
63
 
 
48
  self.conversation_history = []
49
 
50
  def get_response(self, content: str, query: str) -> tuple[Optional[str], Optional[str]]:
51
+ """LLMを使用して回答を生成"""
52
  try:
53
  prompt = self.create_prompt(content, query)
54
  self._add_to_history("user", prompt)
55
 
56
+ # Claudeへのリクエストを修正
57
  response = self.claude_client.messages.create(
58
+ model="claude-3-sonnet-20240229",
59
+ max_tokens=4096,
60
+ messages=[
61
+ {"role": "user", "content": prompt}
62
+ ]
63
  )
 
64
 
65
+ answer = response.content[0].text
66
  self._add_to_history("assistant", answer)
67
  return answer, None
68