Spaces:
Running
Running
aa
Browse files
app.py
CHANGED
@@ -108,28 +108,22 @@ generation_config = types.GenerateContentConfig(
|
|
108 |
model_flash_exp = "gemini-2.0-pro-exp-02-05"
|
109 |
model_gemini_ex = "gemini-2.0-flash-thinking-exp-01-21"
|
110 |
|
111 |
-
# Función
|
112 |
-
async def
|
113 |
-
|
114 |
model=model_name,
|
115 |
-
contents=borrador,
|
116 |
config=generation_config,
|
117 |
)
|
118 |
-
|
119 |
-
async for chunk in response_stream:
|
120 |
-
full_response += chunk.text
|
121 |
-
return full_response
|
122 |
|
123 |
-
# Combina las respuestas de dos modelos
|
124 |
async def combine_responses(borrador):
|
125 |
-
flash_task = asyncio.create_task(
|
126 |
-
gemini_ex_task = asyncio.create_task(
|
127 |
flash_result = await flash_task
|
128 |
gemini_ex_result = await gemini_ex_task
|
129 |
-
combined_result =
|
130 |
-
f"**Google Gemini flash-exp**:\n{flash_result}\n\n"
|
131 |
-
f"**Google gemini-exp-1206**:\n{gemini_ex_result}"
|
132 |
-
)
|
133 |
return combined_result
|
134 |
|
135 |
# Función asíncrona principal de predicción
|
|
|
108 |
model_flash_exp = "gemini-2.0-pro-exp-02-05"
|
109 |
model_gemini_ex = "gemini-2.0-flash-thinking-exp-01-21"
|
110 |
|
111 |
+
# Función ASÍNCRONA para generar contenido (SIN STREAMING)
|
112 |
+
async def generate_content_async(model_name, borrador):
|
113 |
+
response = await client.aio.models.generate_content( # Usamos la versión async de generate_content
|
114 |
model=model_name,
|
115 |
+
contents=borrador,
|
116 |
config=generation_config,
|
117 |
)
|
118 |
+
return response.text
|
|
|
|
|
|
|
119 |
|
120 |
+
# Combina las respuestas de dos modelos (ASÍNCRONO)
|
121 |
async def combine_responses(borrador):
|
122 |
+
flash_task = asyncio.create_task(generate_content_async(model_flash_exp, borrador)) # Llama a la versión async SIN streaming
|
123 |
+
gemini_ex_task = asyncio.create_task(generate_content_async(model_gemini_ex, borrador)) # Llama a la versión async SIN streaming
|
124 |
flash_result = await flash_task
|
125 |
gemini_ex_result = await gemini_ex_task
|
126 |
+
combined_result = flash_result + "\n\n" + gemini_ex_result
|
|
|
|
|
|
|
127 |
return combined_result
|
128 |
|
129 |
# Función asíncrona principal de predicción
|