Floke/Brancheneinstufung2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: Rename call_gemini_flash to call_gemini_api and add temperature parameter

Browse Source
This commit is contained in:
Jarvis
2026-01-31 06:29:56 +00:00
parent 14dd48b22a
commit 0f0033ff9b
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transcription-tool/backend/services/insights_service.py
View File

@@ -95,7 +95,7 @@ def generate_insight(db: Session, meeting_id: int, insight_type: str) -> databas
db.commit() db.commit()
try: try:
generated_text = call_gemini_flash(prompt=final_prompt, temperature=0.5) generated_text = call_gemini_api(prompt=final_prompt, temperature=0.5)
# 5. Store the new insight # 5. Store the new insight
new_insight = database.AnalysisResult( new_insight = database.AnalysisResult(

5
transcription-tool/backend/services/llm_service.py
View File

@@ -6,12 +6,13 @@ import time
# Configure logging # Configure logging
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
def call_gemini_api(prompt: str, retries: int = 3, timeout: int = 600) -> str: def call_gemini_api(prompt: str, temperature: float = 0.7, retries: int = 3, timeout: int = 600) -> str:
""" """
Calls the Gemini Pro API with a given prompt. Calls the Gemini Pro API with a given prompt.
Args: Args:
prompt: The text prompt to send to the API. prompt: The text prompt to send to the API.
temperature: Controls randomness in output.
retries: The number of times to retry on failure. retries: The number of times to retry on failure.
timeout: The request timeout in seconds. timeout: The request timeout in seconds.
@@ -33,7 +34,7 @@ def call_gemini_api(prompt: str, retries: int = 3, timeout: int = 600) -> str:
"parts": [{"text": prompt}] "parts": [{"text": prompt}]
}], }],
"generationConfig": { "generationConfig": {
"temperature": 0.7, "temperature": temperature,
"topK": 40, "topK": 40,
"topP": 0.95, "topP": 0.95,
"maxOutputTokens": 8192, "maxOutputTokens": 8192,