Update app.py

app.py

@@ -66,8 +66,7 @@ class SalesOrder(BaseModel):
 class ClinicalNoteResult(BaseModel):
     entities: list
     message: str
-os.environ['GEMINI_API_KEY'] = 'AIzaSyAsVIHsPIIfDBTb2K6VNdNlMt05t8x3mtE'
-
+    
 # # Create a system prompt to guide the model
 system_prompt="Anda adalah asisten medis yang membantu mengekstrak informasi dari catatan klinis. Gunakan alat extract_entities untuk memproses teks."
 
@@ -76,11 +75,6 @@ system_prompt="Anda adalah asisten medis yang membantu mengekstrak informasi dar
 agent = Agent('gemini-1.5-flash', system_prompt=system_prompt) # OK-Gemini
 #agent = Agent(model) # OK-Lokal 
 
-
-
-#agent = Agent(model, result_type=SalesOrder) # belum bisa untuk local
-#agent = Agent(model, system_prompt=SYSTEM_PROMPT) # belum bisa untuk local
-#agent= Agent('gemini-1.5-flash', system_prompt=SYSTEM_PROMPT) # ERR
 THIS_DIR = Path(__file__).parent
 
 
@@ -116,15 +110,12 @@ async def get_chat(database: Database = Depends(get_db)) -> Response:
         media_type='text/plain',
     )
 
-
-    
 class ChatMessage(TypedDict):
     """Format of messages sent to the browser."""
     role: Literal['user', 'model']
     timestamp: str
     content: str
 
-
 def to_chat_message(m: ModelMessage) -> ChatMessage:
     first_part = m.parts[0]
     if isinstance(m, ModelRequest):
@@ -163,16 +154,6 @@ def to_ds_message(m: ModelMessage) -> ChatMessage:
             }
     raise UnexpectedModelBehavior(f'Unexpected ds-message type for chat app: {m}')
     
-def to_form_message(m: ModelMessage) -> ChatMessage:
-    first_part = m.parts[0]
-    if isinstance(m, ModelResponse):
-        if isinstance(first_part, TextPart):
-            return {
-                'role': 'form',
-                'timestamp': m.timestamp.isoformat(),
-                'content': first_part.content,
-            }
-    raise UnexpectedModelBehavior(f'Unexpected message type for chat app: {m}')
 @app.post('/chat/')
 async def post_chat(
     prompt: Annotated[str, fastapi.Form()], database: Database = Depends(get_db)