Update app.py

app.py

@@ -20,7 +20,7 @@ from requests.adapters import HTTPAdapter, Retry
 from typing import List, Optional, Dict, Any
 
 
-class GitHubModelLLM(BaseChatModel):
+class GitHubModelLLM(LLM):
     """GitHub Models API를 사용하는 ChatOpenAI 대체 클래스"""
 
     model_name: str = "openai/gpt-4.1"
@@ -48,7 +48,7 @@ class GitHubModelLLM(BaseChatModel):
         resp.raise_for_status()
         return resp.json()
 
-    def _generate(self, messages: List[Any], stop: Optional[List[str]] = None, **kwargs):
+    def _call(self, messages: List[Any], stop: Optional[List[str]] = None, **kwargs):
         """ConversationBufferMemory의 이전 대화까지 포함해 메시지 생성"""
         msg_list = []
 
@@ -73,8 +73,8 @@ class GitHubModelLLM(BaseChatModel):
         content = res.get("choices", [{}])[0].get("message", {}).get("content", "")
         return content
 
-    async def _agenerate(self, messages: List[Any], stop: Optional[List[str]] = None, **kwargs):
-        return self._generate(messages, stop, **kwargs)
+    async def _acall(self, messages: List[Any], stop: Optional[List[str]] = None, **kwargs):
+        return self._call(messages, stop, **kwargs)
 from typing import Optional, List, Dict, Any
 from langchain.llms.base import LLM
 import requests, os, json