Remove semantic code intelligence from Python agent

Author: AntoineToussaint

pkg/code/callgraph.go

@@ -5,12 +5,8 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"os"
-	"path/filepath"
-	"strings"
 
 	basev0 "github.com/codefly-dev/core/generated/go/codefly/base/v0"
-	codev0 "github.com/codefly-dev/core/generated/go/codefly/services/code/v0"
 	runners "github.com/codefly-dev/core/runners/base"
 )
 
@@ -176,133 +172,3 @@ print(json.dumps({'calls': calls, 'implements': implements, 'module': workdir}))
 	result.Module = srcDir
 	return result
 }
-
-// handleFindReferences finds all usages of a symbol via grep + AST.
-func (c *Code) handleFindReferences(ctx context.Context, req *codev0.CodeRequest) (*codev0.CodeResponse, error) {
-	c.EnsureInit()
-	srcDir := c.SourceDir()
-
-	refs := req.GetFindReferences()
-	if refs == nil {
-		return &codev0.CodeResponse{
-			Result: &codev0.CodeResponse_FindReferences{
-				FindReferences: &codev0.FindReferencesResponse{},
-			},
-		}, nil
-	}
-
-	symbolName, err := symbolAtPosition(srcDir, refs.File, refs.Line, refs.Column)
-	if err != nil || symbolName == "" {
-		return &codev0.CodeResponse{
-			Result: &codev0.CodeResponse_FindReferences{
-				FindReferences: &codev0.FindReferencesResponse{},
-			},
-		}, nil
-	}
-
-	locations := c.grepForSymbol(ctx, srcDir, symbolName)
-
-	return &codev0.CodeResponse{
-		Result: &codev0.CodeResponse_FindReferences{
-			FindReferences: &codev0.FindReferencesResponse{
-				Locations: locations,
-			},
-		},
-	}, nil
-}
-
-func symbolAtPosition(srcDir, file string, line, col int32) (string, error) {
-	absPath := file
-	if !filepath.IsAbs(file) {
-		absPath = filepath.Join(srcDir, file)
-	}
-	data, err := os.ReadFile(absPath)
-	if err != nil {
-		return "", err
-	}
-	lines := strings.Split(string(data), "\n")
-	if int(line) < 1 || int(line) > len(lines) {
-		return "", fmt.Errorf("line %d out of range", line)
-	}
-	lineText := lines[line-1]
-	if int(col) < 1 || int(col) > len(lineText) {
-		col = 1
-	}
-	start := int(col) - 1
-	end := start
-	for start > 0 && isIdentChar(lineText[start-1]) {
-		start--
-	}
-	for end < len(lineText) && isIdentChar(lineText[end]) {
-		end++
-	}
-	if start >= end {
-		return "", nil
-	}
-	return lineText[start:end], nil
-}
-
-func isIdentChar(c byte) bool {
-	return (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || (c >= '0' && c <= '9') || c == '_'
-}
-
-func (c *Code) grepForSymbol(ctx context.Context, srcDir, symbol string) []*codev0.Location {
-	// Route grep through the plugin's active RunnerEnvironment so a
-	// Docker-mode Python agent searches its own mounted tree. Pre-Init
-	// uses ResolveStandaloneEnvironment which honors declared context.
-	env := c.Service.ActiveEnv
-	if env == nil {
-		var rctx *basev0.RuntimeContext
-		if c.Service.Base != nil && c.Service.Base.Runtime != nil {
-			rctx = c.Service.Base.Runtime.RuntimeContext
-		}
-		env = runners.ResolveStandaloneEnvironment(ctx, srcDir, rctx)
-	}
-	proc, procErr := env.NewProcess("grep", "-rn", "--include=*.py", "-w", symbol, srcDir)
-	if procErr != nil {
-		return nil
-	}
-	var buf bytes.Buffer
-	proc.WithOutput(&buf)
-	_ = proc.Run(ctx) // grep returns non-zero on no match; that's fine
-	out := buf.Bytes()
-
-	var locations []*codev0.Location
-	for _, line := range strings.Split(string(out), "\n") {
-		line = strings.TrimSpace(line)
-		if line == "" {
-			continue
-		}
-		parts := strings.SplitN(line, ":", 3)
-		if len(parts) < 2 {
-			continue
-		}
-		filePath := parts[0]
-		relPath, _ := filepath.Rel(srcDir, filePath)
-		if relPath == "" {
-			relPath = filePath
-		}
-
-		lineNum := int32(0)
-		fmt.Sscanf(parts[1], "%d", &lineNum)
-
-		col := int32(1)
-		if len(parts) >= 3 {
-			idx := strings.Index(parts[2], symbol)
-			if idx >= 0 {
-				col = int32(idx + 1)
-			}
-		}
-
-		locations = append(locations, &codev0.Location{
-			File:   relPath,
-			Line:   lineNum,
-			Column: col,
-		})
-
-		if len(locations) >= 100 {
-			break
-		}
-	}
-	return locations
-}

pkg/code/code.go

@@ -17,13 +17,8 @@ import (
 // It embeds PythonCodeServer from core, which provides:
 //   - File ops: ReadFile, WriteFile, CreateFile, DeleteFile, MoveFile, ListFiles, Search
 //   - Git ops: GitLog, GitDiff, GitShow, GitBlame
-//   - Analysis: ListSymbols (AST), GetProjectInfo, ListDependencies
+//   - Analysis: GetProjectInfo, ListDependencies
 //   - Smart edit: ApplyEdit (fuzzy find/replace)
-//
-// Python-specific overrides added here:
-//   - find_references: grep + AST-based symbol reference finding
-//   - get_call_graph: served via Tooling gRPC (Tooling.GetCallGraph) —
-//     the Execute Override path does NOT work for get_call_graph.
 type Code struct {
 	*corecode.PythonCodeServer
 	Service *pythonservice.Service
@@ -39,7 +34,6 @@ func New(svc *pythonservice.Service) *Code {
 		Service:          svc,
 		PythonCodeServer: corecode.NewPythonCodeServer(".", nil),
 	}
-	c.registerOverrides()
 	return c
 }
 
@@ -61,30 +55,17 @@ func (c *Code) SourceDir() string {
 func (c *Code) EnsureInit() {
 	if !c.initialized {
 		c.PythonCodeServer = corecode.NewPythonCodeServer(c.SourceDir(), nil)
-		c.registerOverrides()
 		c.initialized = true
 	}
 }
 
-// registerOverrides wires Python-specific handlers on top of PythonCodeServer.
-// PythonCodeServer already provides: list_symbols (AST), get_project_info, list_dependencies.
-// Specializations may add their own via c.Override(...) after constructing.
-func (c *Code) registerOverrides() {
-	c.Override("find_references", c.handleFindReferences)
-}
-
 // ── Lazy-init RPC wrappers ───────────────────────────────
 
 func (c *Code) GetProjectInfo(ctx context.Context, req *codev0.GetProjectInfoRequest) (*codev0.GetProjectInfoResponse, error) {
 	c.EnsureInit()
 	return c.PythonCodeServer.GetProjectInfo(ctx, req)
 }
 
-func (c *Code) ListSymbols(ctx context.Context, req *codev0.ListSymbolsRequest) (*codev0.ListSymbolsResponse, error) {
-	c.EnsureInit()
-	return c.PythonCodeServer.ListSymbols(ctx, req)
-}
-
 func (c *Code) ListDependencies(ctx context.Context, req *codev0.ListDependenciesRequest) (*codev0.ListDependenciesResponse, error) {
 	c.EnsureInit()
 	return c.PythonCodeServer.ListDependencies(ctx, req)