Upload 5 files

app.py

@@ -1,6 +1,6 @@
 """
-Jade Code IDE - HuggingFace Space Backend v3.0
-Modo Agêntico + Multi-Provider (Groq, Cerebras, OpenRouter)
+Jade Code IDE - HuggingFace Space Backend v3.1
+Modo Agêntico Melhorado + Context-Aware + Multi-Provider
 """
 
 import os
@@ -18,11 +18,11 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import gradio as gr
 
-from providers import get_provider, list_all_providers, GroqProvider, CerebrasProvider, OpenRouterProvider
+from providers import get_provider, list_all_providers
 from tools import AGENT_TOOLS, ToolExecutor, parse_tool_calls, get_response_content, has_tool_calls
 
 # ============== App Setup ==============
-app = FastAPI(title="Jade Code IDE API", version="3.0")
+app = FastAPI(title="Jade Code IDE API", version="3.1")
 
 app.add_middleware(
     CORSMiddleware,
@@ -40,7 +40,7 @@ class ChatRequest(BaseModel):
     model: str = "llama-3.3-70b-versatile"
     files: Optional[Dict[str, str]] = {}  # {filename: content}
     current_file: Optional[str] = None
-    agentic: bool = True  # Modo agêntico ativado por padrão
+    agentic: bool = True
     history: Optional[List[Dict]] = []
 
 class RunRequest(BaseModel):
@@ -54,29 +54,76 @@ class CompleteRequest(BaseModel):
     model: str = "llama-3.3-70b-versatile"
 
 
+# ============== Helper: Build Context ==============
+def build_file_context(files: Dict[str, str], current_file: Optional[str], max_chars: int = 8000) -> str:
+    """
+    Constrói contexto dos arquivos para o LLM.
+    Inclui conteúdo resumido de todos os arquivos.
+    """
+    if not files:
+        return ""
+    
+    context_parts = []
+    total_chars = 0
+    
+    # Lista de arquivos
+    context_parts.append(f"📁 **PROJETO ({len(files)} arquivos)**")
+    context_parts.append("```")
+    for name in files.keys():
+        is_current = " ← [ABERTO]" if name == current_file else ""
+        context_parts.append(f"  {name}{is_current}")
+    context_parts.append("```\n")
+    
+    # Conteúdo do arquivo atual (prioridade máxima)
+    if current_file and current_file in files:
+        content = files[current_file]
+        if len(content) > 4000:
+            content = content[:4000] + f"\n... [truncado, {len(content)} chars total]"
+        context_parts.append(f"📄 **ARQUIVO ABERTO: {current_file}**")
+        context_parts.append(f"```python\n{content}\n```\n")
+        total_chars += len(content)
+    
+    # Outros arquivos (resumido)
+    for name, content in files.items():
+        if name == current_file:
+            continue
+        if total_chars > max_chars:
+            context_parts.append(f"_... mais {len(files) - len(context_parts) + 5} arquivos não mostrados_")
+            break
+        
+        # Mostra preview pequeno
+        preview = content[:500] if len(content) > 500 else content
+        if len(content) > 500:
+            preview += f"\n... [{len(content)} chars total]"
+        
+        context_parts.append(f"📄 **{name}**")
+        ext = name.split('.')[-1] if '.' in name else 'text'
+        lang = {'py': 'python', 'js': 'javascript', 'md': 'markdown', 'json': 'json'}.get(ext, ext)
+        context_parts.append(f"```{lang}\n{preview}\n```\n")
+        total_chars += len(preview)
+    
+    return "\n".join(context_parts)
+
+
 # ============== System Prompt ==============
-SYSTEM_PROMPT = """Você é CodeJade, um assistente de programação avançado integrado a uma IDE.
-
-CAPACIDADES:
-- Você pode explorar o projeto do usuário usando ferramentas
-- Você tem acesso aos arquivos do projeto
-- Você pode ajudar a escrever, debugar e explicar código
-
-FERRAMENTAS DISPONÍVEIS:
-1. list_files - Lista todos os arquivos do projeto
-2. read_file - Lê o conteúdo de um arquivo específico
-3. search_code - Busca texto nos arquivos do projeto
-
-REGRAS:
-- Use as ferramentas quando precisar entender o contexto do projeto
-- Seja direto e técnico
-- Quando sugerir código, use blocos ```python
-- Explique o que está fazendo
-
-EXEMPLO:
-Usuário: "Me explica o que esse projeto faz"
-Você: [usa list_files para ver estrutura, depois read_file nos arquivos importantes]
-Depois explica baseado no que leu.
+SYSTEM_PROMPT = """Você é **CodeJade**, um assistente de programação integrado a uma IDE web.
+
+## Suas Capacidades:
+- Você tem acesso COMPLETO ao projeto do usuário (arquivos listados abaixo)
+- Você pode ver o arquivo que está aberto no editor
+- Você ajuda a escrever código, debugar, explicar e melhorar
+
+## Regras:
+1. LEIA o contexto do projeto antes de responder
+2. Seja técnico e direto
+3. Use blocos ```python para código
+4. Se for modificar código, mostre apenas as partes que mudam
+5. Considere TODO o projeto, não só o arquivo aberto
+
+## Formato de Resposta:
+- Para explicações: texto normal com formatação markdown
+- Para código: blocos de código com linguagem especificada
+- Para bugs: mostre o problema E a solução
 """
 
 
@@ -84,20 +131,17 @@ Depois explica baseado no que leu.
 
 @app.get("/")
 async def root():
-    return {"status": "ok", "message": "Jade Code IDE API v3.0 - Agentic Mode", "version": "3.0"}
+    return {"status": "ok", "message": "Jade Code IDE API v3.1 - Context-Aware Mode", "version": "3.1"}
 
 
 @app.get("/providers")
 async def get_providers():
-    """Lista providers disponíveis."""
     return {"providers": list_all_providers()}
 
 
 @app.get("/models/{provider}")
 async def get_models(provider: str):
-    """Lista modelos de um provider."""
     try:
-        # Cria provider com key dummy só pra pegar os modelos
         prov = get_provider(provider, "dummy")
         return {"models": prov.list_models()}
     except ValueError as e:
@@ -107,83 +151,86 @@ async def get_models(provider: str):
 @app.post("/chat")
 async def chat(req: ChatRequest, x_api_key: str = Header(..., alias="X-API-Key")):
     """
-    Chat com modo agêntico.
-    O LLM pode usar ferramentas para explorar o projeto.
+    Chat com contexto completo do projeto.
+    Funciona com ou sem suporte a tool calls.
     """
     if not x_api_key or len(x_api_key) < 10:
         raise HTTPException(status_code=401, detail="API Key inválida")
     
     try:
         provider = get_provider(req.provider, x_api_key)
-        tool_executor = ToolExecutor(req.files)
         
-        # Monta mensagens iniciais
-        messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+        # Constrói contexto dos arquivos
+        file_context = build_file_context(req.files, req.current_file)
         
-        # Adiciona histórico
-        for msg in (req.history or [])[-10:]:
-            messages.append(msg)
+        # System prompt com contexto
+        system_with_context = SYSTEM_PROMPT
+        if file_context:
+            system_with_context += f"\n\n---\n\n# Contexto do Projeto\n\n{file_context}"
         
-        # Contexto do arquivo atual
-        context_parts = []
-        if req.files:
-            context_parts.append(f"📁 Projeto tem {len(req.files)} arquivo(s): {', '.join(req.files.keys())}")
-        if req.current_file and req.current_file in (req.files or {}):
-            content = req.files[req.current_file][:2000]
-            context_parts.append(f"📄 Arquivo aberto ({req.current_file}):\n```\n{content}\n```")
+        # Monta mensagens
+        messages = [{"role": "system", "content": system_with_context}]
         
-        user_msg = req.message
-        if context_parts:
-            user_msg = "\n".join(context_parts) + f"\n\n💬 Mensagem: {req.message}"
+        # Histórico
+        for msg in (req.history or [])[-6:]:
+            messages.append(msg)
         
-        messages.append({"role": "user", "content": user_msg})
+        # Mensagem do usuário
+        messages.append({"role": "user", "content": req.message})
         
-        # Loop agêntico
-        max_iterations = 5
+        # Tenta com tools primeiro (se agentic)
         tool_results = []
         
-        for i in range(max_iterations):
-            # Chama LLM
-            if req.agentic:
+        if req.agentic:
+            try:
+                # Tenta chamada com tools
+                tool_executor = ToolExecutor(req.files)
                 response = provider.chat(messages, req.model, tools=AGENT_TOOLS)
-            else:
-                response = provider.chat(messages, req.model)
-            
-            # Verifica se tem tool calls
-            if has_tool_calls(response):
-                tool_calls = parse_tool_calls(response)
                 
-                # Adiciona resposta do assistente com tool calls
-                assistant_msg = response["choices"][0]["message"]
-                messages.append(assistant_msg)
+                # Se tiver tool calls, executa
+                if has_tool_calls(response):
+                    for _ in range(3):  # Max 3 iterações
+                        tool_calls = parse_tool_calls(response)
+                        if not tool_calls:
+                            break
+                        
+                        assistant_msg = response["choices"][0]["message"]
+                        messages.append(assistant_msg)
+                        
+                        for tc in tool_calls:
+                            result = tool_executor.execute(tc["name"], tc["args"])
+                            tool_results.append({"tool": tc["name"], "result": result[:300]})
+                            messages.append({
+                                "role": "tool",
+                                "tool_call_id": tc["id"],
+                                "content": result
+                            })
+                        
+                        response = provider.chat(messages, req.model, tools=AGENT_TOOLS)
+                        if not has_tool_calls(response):
+                            break
                 
-                # Executa cada tool
-                for tc in tool_calls:
-                    result = tool_executor.execute(tc["name"], tc["args"])
-                    tool_results.append({"tool": tc["name"], "result": result[:500]})
-                    
-                    # Adiciona resultado como mensagem de tool
-                    messages.append({
-                        "role": "tool",
-                        "tool_call_id": tc["id"],
-                        "content": result
-                    })
-                
-                # Continua o loop
-                continue
-            else:
-                # Resposta final
                 content = get_response_content(response)
                 return {
                     "response": content,
                     "tool_calls": tool_results if tool_results else None
                 }
-        
-        # Se chegou aqui, atingiu limite
-        return {
-            "response": "⚠️ Limite de iterações atingido. Tente uma pergunta mais específica.",
-            "tool_calls": tool_results
-        }
+            
+            except Exception as tool_error:
+                # Fallback: chamada sem tools (modelo não suporta)
+                print(f"Tool call failed, falling back: {tool_error}")
+                response = provider.chat(messages, req.model)
+                content = get_response_content(response)
+                return {
+                    "response": content,
+                    "tool_calls": None,
+                    "note": "Modelo não suporta tools, usando contexto direto"
+                }
+        else:
+            # Modo sem tools
+            response = provider.chat(messages, req.model)
+            content = get_response_content(response)
+            return {"response": content}
     
     except Exception as e:
         traceback.print_exc()
@@ -259,22 +306,20 @@ async def autocomplete(req: CompleteRequest, x_api_key: str = Header(..., alias=
     try:
         provider = get_provider(req.provider, x_api_key)
         
-        prompt = f"""Você é um autocomplete de código. Sugira completions.
+        prompt = f"""Complete o código. Responda APENAS JSON:
 
-CÓDIGO ANTES DO CURSOR:
+ANTES DO CURSOR:
 ```{req.language}
-{req.prefix[-500:]}
+{req.prefix[-400:]}
 ```
 
-CÓDIGO DEPOIS DO CURSOR:
+DEPOIS:
 ```{req.language}
-{req.suffix[:200] if req.suffix else ''}
+{req.suffix[:150] if req.suffix else ''}
 ```
 
-Responda APENAS com um JSON array:
-[{{"label": "nome", "insertText": "código", "detail": "descrição", "kind": "Function"}}]
-
-Máximo 5 sugestões. Tipos: Function, Variable, Class, Module, Keyword, Snippet"""
+JSON (max 5):
+[{{"label": "nome", "insertText": "código", "detail": "desc", "kind": "Function"}}]"""
 
         response = provider.chat(
             messages=[
@@ -317,8 +362,8 @@ def gradio_chat(message, provider, model, api_key):
 
 
 with gr.Blocks(title="Jade Code IDE API") as demo:
-    gr.Markdown("# 🟢 Jade Code IDE - API Backend v3.0")
-    gr.Markdown("**Modo Agêntico** | Multi-Provider | API Docs: `/docs`")
+    gr.Markdown("# 🟢 Jade Code IDE - API Backend v3.1")
+    gr.Markdown("**Context-Aware Mode** | Multi-Provider | API Docs: `/docs`")
     
     with gr.Row():
         provider_dd = gr.Dropdown(
@@ -327,7 +372,7 @@ with gr.Blocks(title="Jade Code IDE API") as demo:
             label="Provider"
         )
         model_dd = gr.Dropdown(
-            choices=["llama-3.3-70b-versatile", "llama-3.1-8b-instant"],
+            choices=["llama-3.3-70b-versatile"],
             value="llama-3.3-70b-versatile",
             label="Model"
         )