Spaces:

habulaj
/

newapi-clone

Running

App Files Files Community

habulaj commited on 2 days ago

Commit

ae19f3c

verified ·

1 Parent(s): 75bb18b

Update routers/inference.py

Browse files

Files changed (1) hide show

routers/inference.py +64 -26

routers/inference.py CHANGED Viewed

@@ -169,10 +169,16 @@ def extract_text_from_response(response):
     """
     logger.info(f"Tipo da resposta: {type(response)}")
-    # Método 1: Verificar se tem atributo 'text' diretamente
-    if hasattr(response, 'text') and response.text:
-        logger.info("Texto extraído via response.text")
-        return response.text
     # Método 2: Verificar candidates
     if hasattr(response, 'candidates') and response.candidates:
@@ -188,33 +194,65 @@ def extract_text_from_response(response):
                 # Verificar se tem parts
                 if hasattr(content, 'parts') and content.parts:
-                    logger.info(f"Content tem {len(list(content.parts))} parts")
-                    response_text = ""
-                    for j, part in enumerate(content.parts):
-                        logger.info(f"Processando part {j}, tipo: {type(part)}")
-                        if hasattr(part, 'text') and part.text:
-                            logger.info(f"Part {j} tem texto: {len(part.text)} caracteres")
-                            response_text += part.text
-                    if response_text:
-                        return response_text
-    # Método 3: Tentar acessar via string conversion
     try:
-        response_str = str(response)
-        logger.info(f"Resposta como string (primeiros 200 chars): {response_str[:200]}")
-        # Se a string contém as tags esperadas, retornar ela
-        if '<headline>' in response_str and '<body>' in response_str:
-            return response_str
     except Exception as e:
-        logger.error(f"Erro ao converter resposta para string: {e}")
-    # Debug: Logar atributos da resposta
-    logger.info(f"Atributos da resposta: {dir(response)}")
     return ""

     """
     logger.info(f"Tipo da resposta: {type(response)}")
+    # Método 1: Tentar acessar response.text diretamente
+    try:
+        text_content = getattr(response, 'text', None)
+        if text_content:
+            logger.info(f"Texto extraído via response.text: {len(text_content)} caracteres")
+            return text_content
+        else:
+            logger.info("response.text existe mas está vazio/None")
+    except Exception as e:
+        logger.error(f"Erro ao acessar response.text: {e}")
     # Método 2: Verificar candidates
     if hasattr(response, 'candidates') and response.candidates:
                 # Verificar se tem parts
                 if hasattr(content, 'parts') and content.parts:
+                    try:
+                        parts_list = list(content.parts)
+                        logger.info(f"Content tem {len(parts_list)} parts")
+                        response_text = ""
+                        for j, part in enumerate(parts_list):
+                            logger.info(f"Processando part {j}, tipo: {type(part)}")
+                            # Tentar várias formas de acessar o texto
+                            part_text = None
+                            if hasattr(part, 'text'):
+                                part_text = getattr(part, 'text', None)
+                            if part_text:
+                                logger.info(f"Part {j} tem texto: {len(part_text)} caracteres")
+                                response_text += part_text
+                            else:
+                                logger.info(f"Part {j} não tem texto ou está vazio")
+                        if response_text:
+                            return response_text
+                    except Exception as e:
+                        logger.error(f"Erro ao processar parts do candidate {i}: {e}")
+    # Método 3: Tentar usar método _get_text() se existir
     try:
+        if hasattr(response, '_get_text'):
+            text_content = response._get_text()
+            if text_content:
+                logger.info(f"Texto extraído via _get_text(): {len(text_content)} caracteres")
+                return text_content
     except Exception as e:
+        logger.error(f"Erro ao usar _get_text(): {e}")
+    # Método 4: Debug - tentar inspecionar a estrutura real
+    try:
+        logger.info("Tentando debug da estrutura:")
+        if hasattr(response, 'candidates') and response.candidates:
+            candidate = response.candidates[0]
+            logger.info(f"Primeiro candidate: {type(candidate)}")
+            logger.info(f"Atributos do candidate: {dir(candidate)}")
+            if hasattr(candidate, 'content'):
+                content = candidate.content
+                logger.info(f"Content: {type(content)}")
+                logger.info(f"Atributos do content: {dir(content)}")
+                if hasattr(content, 'parts'):
+                    logger.info(f"Parts: {type(content.parts)}")
+                    try:
+                        parts_list = list(content.parts)
+                        if parts_list:
+                            first_part = parts_list[0]
+                            logger.info(f"Primeiro part: {type(first_part)}")
+                            logger.info(f"Atributos do part: {dir(first_part)}")
+                    except Exception as e:
+                        logger.error(f"Erro ao inspecionar parts: {e}")
+    except Exception as e:
+        logger.error(f"Erro no debug da estrutura: {e}")
     return ""