fix: Use entire file as context in "Optimize Code" prompt

buf.gen.yaml

@@ -9,16 +9,15 @@ plugins:
     out: src/shared/pyroscope-api
     opt: target=ts
 inputs:
-    # Build a subset of the Pyroscope protobuf API
+  # Build a subset of the Pyroscope protobuf API
   - git_repo: https://github.com/grafana/pyroscope.git
     # Update this to any commit, which is merged in Pyroscope main
     ref: weekly-f105-886b418af
     subdir: api
     paths:
-    - adhocprofiles/
-    - querier/
-    - settings/
-    - types/
-    - vcs/
-    - google/
-
+      - adhocprofiles/
+      - querier/
+      - settings/
+      - types/
+      - vcs/
+      - google/

src/pages/ProfilesExplorerView/components/SceneExploreServiceFlameGraph/components/SceneFunctionDetailsPanel/components/CodeContainer/CodeContainer.tsx

@@ -22,7 +22,7 @@ export function CodeContainer({ dataSourceUid, functionDetails }: CodeContainerP
   return (
     <>
       <Code
-        lines={data.lines}
+        lines={data.snippetLines}
         unit={data.unit}
         githubUrl={data.githubUrl}
         isLoadingCode={data.isLoadingCode}
@@ -39,7 +39,7 @@ export function CodeContainer({ dataSourceUid, functionDetails }: CodeContainerP
         <AiSuggestionsPanel
           suggestionPromptInputs={{
             functionDetails: functionDetails,
-            lines: data.lines,
+            lines: data.allLines,
           }}
         />
       ) : null}