feat: reimplemented heartbeat button

Vali-98 · Dec 23, 2024 · f7232bf · f7232bf
1 parent 4ef0c13
commit f7232bf
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/app/components/APIManager/AddAPI.tsx b/app/components/APIManager/AddAPI.tsx
@@ -1,5 +1,6 @@
 import ButtonPrimary from '@components/Buttons/ButtonPrimary'
 import DropdownSheet from '@components/DropdownSheet'
+import HeartbeatButton from '@components/Endpoint/HeartbeatButton'
 import MultiDropdownSheet from '@components/MultiDropdownSheet'
 import claudeModels from '@constants/API/ClaudeModels.json'
 import { MaterialIcons } from '@expo/vector-icons'
@@ -113,6 +114,18 @@ const AddAPI = () => {
                             placeholder={template.defaultValues.modelEndpoint}
                             placeholderTextColor={Style.getColor('primary-text2')}
                         />
+                        <HeartbeatButton
+                            api={values.modelEndpoint ?? ''}
+                            apiFormat={(s) => s}
+                            headers={
+                                template.features.useKey
+                                    ? {
+                                          [template.request.authHeader]:
+                                              template.request.authPrefix + values.key,
+                                      }
+                                    : {}
+                            }
+                        />
                     </View>
                 )}
 

diff --git a/app/components/APIManager/EditAPIModal.tsx b/app/components/APIManager/EditAPIModal.tsx
@@ -1,5 +1,6 @@
 import ButtonPrimary from '@components/Buttons/ButtonPrimary'
 import DropdownSheet from '@components/DropdownSheet'
+import HeartbeatButton from '@components/Endpoint/HeartbeatButton'
 import FadeBackrop from '@components/FadeBackdrop'
 import MultiDropdownSheet from '@components/MultiDropdownSheet'
 import { APIConfiguration } from '@constants/API/APIBuilder.types'
@@ -140,6 +141,18 @@ const EditAPIModal: React.FC<EditAPIModalProps> = ({ index, show, close, origina
                                 placeholder={template.defaultValues.modelEndpoint}
                                 placeholderTextColor={Style.getColor('primary-text2')}
                             />
+                            <HeartbeatButton
+                                api={values.modelEndpoint ?? ''}
+                                apiFormat={(s) => s}
+                                headers={
+                                    template.features.useKey
+                                        ? {
+                                              [template.request.authHeader]:
+                                                  template.request.authPrefix + values.key,
+                                          }
+                                        : {}
+                                }
+                            />
                         </View>
                     )}