From 68e138f93117fa3656336013e0f6b225a3da2e03 Mon Sep 17 00:00:00 2001
From: danielcampagnolitg
 <138441775+danielcampagnolitg@users.noreply.github.com>
Date: Tue, 5 Nov 2024 10:53:43 +0800
Subject: [PATCH] filter agent by userId

---
 .../assistants/assistant/assistant.component.ts        | 10 +++++-----
 src/modules/firestore/firestoreAgentStateService.ts    |  2 ++
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/frontend/src/app/modules/assistants/assistant/assistant.component.ts b/frontend/src/app/modules/assistants/assistant/assistant.component.ts
index f24d48e7..9d3fb076 100644
--- a/frontend/src/app/modules/assistants/assistant/assistant.component.ts
+++ b/frontend/src/app/modules/assistants/assistant/assistant.component.ts
@@ -18,13 +18,13 @@ import { MatIconModule } from '@angular/material/icon';
 import { MatInputModule } from '@angular/material/input';
 import { MatMenuModule } from '@angular/material/menu';
 import { MatSidenavModule } from '@angular/material/sidenav';
-import {Router, RouterLink, RouterModule, ActivatedRoute} from '@angular/router';
+import { Router, RouterLink, RouterModule, ActivatedRoute } from '@angular/router';
 import { FuseMediaWatcherService } from '@fuse/services/media-watcher';
 import { AssistantsService } from 'app/modules/assistants/assistants.service';
 import { AssistantChat } from '../assistant.types';
 import { AssistantInfoComponent } from 'app/modules/assistants/assistant-info/assistant-info.component';
 import { LLM, LlmService } from "app/modules/agents/services/llm.service";
-import {BehaviorSubject, Subject, takeUntil} from 'rxjs';
+import { BehaviorSubject, Subject, takeUntil } from 'rxjs';
 import {
     CLIPBOARD_OPTIONS,
     ClipboardButtonComponent,
@@ -32,9 +32,9 @@ import {
     MarkdownService,
     provideMarkdown
 } from "ngx-markdown";
-import {MatOption} from "@angular/material/core";
-import {MatSelect} from "@angular/material/select";
-import {ReactiveFormsModule} from "@angular/forms";
+import { MatOption } from "@angular/material/core";
+import { MatSelect } from "@angular/material/select";
+import { ReactiveFormsModule } from "@angular/forms";
 
 
 @Component({
diff --git a/src/modules/firestore/firestoreAgentStateService.ts b/src/modules/firestore/firestoreAgentStateService.ts
index 9ee78b4c..b24a221e 100644
--- a/src/modules/firestore/firestoreAgentStateService.ts
+++ b/src/modules/firestore/firestoreAgentStateService.ts
@@ -6,6 +6,7 @@ import { AgentStateService } from '#agent/agentStateService/agentStateService';
 import { functionFactory } from '#functionSchema/functionDecorators';
 import { logger } from '#o11y/logger';
 import { span } from '#o11y/trace';
+import { currentUser } from '#user/userService/userContext';
 import { firestoreDb } from './firestore';
 
 /**
@@ -52,6 +53,7 @@ export class FirestoreAgentStateService implements AgentStateService {
 		const keys: Array<keyof AgentContext> = ['agentId', 'name', 'state', 'cost', 'error', 'lastUpdate', 'userPrompt', 'inputPrompt'];
 		const querySnapshot = await this.db
 			.collection('AgentContext')
+			.where('userId', '==', currentUser().id)
 			.select(...keys)
 			.orderBy('lastUpdate', 'desc')
 			.get();