filter agent by userId

TrafficGuard · Nov 5, 2024 · 68e138f · 68e138f
1 parent 10a46ec
commit 68e138f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/frontend/src/app/modules/assistants/assistant/assistant.component.ts b/frontend/src/app/modules/assistants/assistant/assistant.component.ts
@@ -18,23 +18,23 @@ 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,
     MarkdownModule,
     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
@@ -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();