fix(chat): no longer show embeddings in ai model select

DashHub-ai · Dec 26, 2024 · 7d80c7f · 7d80c7f
1 parent e5a7d73
commit 7d80c7f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/apps/backend/src/modules/ai-models/elasticsearch/ai-models-es-search.repo.ts b/apps/backend/src/modules/ai-models/elasticsearch/ai-models-es-search.repo.ts
@@ -86,13 +86,15 @@ export class AIModelsEsSearchRepo {
       ids,
       organizationIds,
       archived,
+      embedding,
     }: SdKSearchAIModelsInputT,
   ): esb.Query =>
     esb.boolQuery().must(
       rejectFalsyItems([
         !!ids?.length && esb.termsQuery('id', ids),
         !!organizationIds?.length && esb.termsQuery('organization.id', organizationIds),
         !isNil(isDefault) && esb.termQuery('default', isDefault),
+        !isNil(embedding) && esb.termQuery('embedding', embedding),
         !!phrase && (
           esb
             .boolQuery()

diff --git a/apps/chat/src/modules/ai-models/controls/ai-models-search-select.tsx b/apps/chat/src/modules/ai-models/controls/ai-models-search-select.tsx
@@ -12,6 +12,7 @@ const AIModelsSearchAbstractSelect = createSdkAutocomplete<
   fetchFn: ({ sdk: { sdks }, phrase, limit, filters }) =>
     sdks.dashboard.aiModels.search({
       archived: false,
+      embedding: false,
       sort: 'score:desc',
       offset: 0,
       limit,

diff --git a/apps/chat/src/modules/chats/start-chat/start-chat-form.tsx b/apps/chat/src/modules/chats/start-chat/start-chat-form.tsx
@@ -61,7 +61,7 @@ export function StartChatForm({ forceProject, className }: Props) {
     >
       <div
         className={clsx(
-          'relative z-10 bg-background shadow-sm border border-border rounded-lg overflow-hidden',
+          'relative z-10 bg-background shadow-sm border border-border rounded-lg',
           'focus-within:border-primary/50',
           'transition-border duration-100',
         )}
@@ -70,7 +70,7 @@ export function StartChatForm({ forceProject, className }: Props) {
           <textarea
             ref={focusInputRef}
             name="message"
-            className="p-4 pb-0 w-full focus:outline-none resize-none"
+            className="p-4 pb-0 rounded-lg w-full focus:outline-none resize-none"
             placeholder={t.placeholder}
             required
             {...bind.path('content')}