2 files changed, 16 insertions, 1 deletions
diff --git a/packages/server/src/config/config.ts b/packages/server/src/config/config.ts
index 41409f7c..daff1807 100644
--- a/packages/server/src/config/config.ts
+++ b/packages/server/src/config/config.ts
@@ -34,6 +34,7 @@ export class Config {
     private readonly toolDiscoveryCommand: string | undefined,
     private readonly toolCallCommand: string | undefined,
     private readonly mcpServerCommand: string | undefined,
+    private readonly userAgent: string,
   ) {
     // toolRegistry still needs initialization based on the instance
     this.toolRegistry = createToolRegistry(this);
@@ -81,6 +82,10 @@ export class Config {
   getMcpServerCommand(): string | undefined {
     return this.mcpServerCommand;
   }
+
+  getUserAgent(): string {
+    return this.userAgent;
+  }
 }
 
 function findEnvFile(startDir: string): string | null {
@@ -117,6 +122,7 @@ export function createServerConfig(
   toolDiscoveryCommand?: string,
   toolCallCommand?: string,
   mcpServerCommand?: string,
+  userAgent?: string,
 ): Config {
   return new Config(
     apiKey,
@@ -129,6 +135,7 @@ export function createServerConfig(
     toolDiscoveryCommand,
     toolCallCommand,
     mcpServerCommand,
+    userAgent ?? 'GeminiCLI/unknown', // Default user agent
   );
 }
 
diff --git a/packages/server/src/core/client.ts b/packages/server/src/core/client.ts
index b8d47476..0b1e8ecf 100644
--- a/packages/server/src/core/client.ts
+++ b/packages/server/src/core/client.ts
@@ -34,7 +34,15 @@ export class GeminiClient {
   private readonly MAX_TURNS = 100;
 
   constructor(private config: Config) {
-    this.client = new GoogleGenAI({ apiKey: config.getApiKey() });
+    const userAgent = config.getUserAgent();
+    this.client = new GoogleGenAI({
+      apiKey: config.getApiKey(),
+      httpOptions: {
+        headers: {
+          'User-Agent': userAgent,
+        },
+      },
+    });
     this.model = config.getModel();
   }