diff --git a/services/useApiService.ts b/services/useApiService.ts
index 79ad3fc..5e7e717 100644
--- a/services/useApiService.ts
+++ b/services/useApiService.ts
@@ -2,12 +2,14 @@ import { useCallback } from 'react';
 
 import { useFetch } from '@/hooks/useFetch';
 
+import {OllamaModel, OllamaModelDetail} from '@/types/ollama'
+
 
 const useApiService = () => {
   const fetchService = useFetch();
 
   const getModels = useCallback(
-    () => {
+    (): Promise<OllamaModel[]>  => {
       return fetchService.get(`/api/models`, {
         headers: {
           'Content-Type': 'application/json',