fix: get the GitHub avatar from the api instead of the cache (#561)

* feat: add the api to get the github avatar

* fix: close the issue #554

client/app/factory/edit/components/BotCreateForm.tsx

@@ -24,7 +24,7 @@ import InputList from './InputList';
 import BulbIcon from '@/public/icons/BulbIcon';
 import GitHubIcon from '@/public/icons/GitHubIcon';
 import { random } from 'lodash';
-import { useBotDelete } from '@/app/hooks/useBot';
+import { useBotDelete, useGetGitAvatar } from '@/app/hooks/useBot';
 import { ToastContainer, toast } from 'react-toastify';
 import { useBot } from '@/app/contexts/BotContext';
 
@@ -37,6 +37,7 @@ import CreateButton from '@/app/user/tokens/components/CreateButton';
 
 const BotCreateFrom = () => {
   const { botProfile, setBotProfile } = useBot();
+  const { data: gitAvatar } = useGetGitAvatar(botProfile?.repoName);
   const router = useRouter();
   const { isOpen, onOpen, onOpenChange, onClose } = useDisclosure();
   const { data: availableLLMs = [] } = useAvailableLLMs();
@@ -128,7 +129,7 @@ const BotCreateFrom = () => {
                   aria-label={I18N.components.BotCreateFrom.gITHU}
                   onClick={() => {
                     setBotProfile((draft: BotProfile) => {
-                      draft.avatar = botProfile?.gitAvatar;
+                      draft.avatar = gitAvatar;
                     });
                   }}
                 >

client/app/hooks/useBot.ts

@@ -9,6 +9,7 @@ import {
   getBotInfoByRepoName,
   getBotList,
   getChunkList,
+  getGitAvatarByRepoName,
   getRagTask,
   getUserPeterCatAppRepos,
   publicBot,
@@ -191,7 +192,16 @@ export const useGetUserPeterCatAppRepos = (enabled: boolean = true) => {
     queryKey: ['github.user.app.repos'],
     queryFn: async () => getUserPeterCatAppRepos(),
     select: (data) => data.data,
-    enabled
+    enabled,
+  });
+};
+
+export const useGetGitAvatar = (repoName?: string) => {
+  return useQuery({
+    queryKey: ['github.repo.name', repoName],
+    queryFn: async () => getGitAvatarByRepoName(repoName!),
+    select: (data) => data.data.data,
+    enabled: !!repoName,
   });
 };
 

client/app/services/BotsController.ts

@@ -67,6 +67,10 @@ export async function getBotInfoByRepoName(params: {
   return axios.post(`${apiDomain}/api/bot/config/generator`, params);
 }
 
+export async function getGitAvatarByRepoName(repo_name: string) {
+  return axios.get(`${apiDomain}/api/bot/git/avatar?repo_name=${repo_name}`);
+}
+
 export async function getChunkList(
   repo_name: string,
   page_size: number,

server/bot/router.py

@@ -173,6 +173,22 @@ async def bot_generator(
             content={"success": False, "errorMessage": str(e)}, status_code=500
         )
 
+@router.get("/git/avatar", status_code=200)
+async def get_git_avatar(
+    repo_name: str,
+):
+    try:
+        g = Github()
+        repo = g.get_repo(repo_name)
+        avatar = repo.organization.avatar_url if repo.organization else None
+        return JSONResponse(content={"success": True, "data": avatar})
+    except Exception as e:
+        return JSONResponse(
+            content={"success": False, "errorMessage": str(e)}, status_code=500
+        )
+
+
+
 
 @router.put("/update/{id}", status_code=200)
 def update_bot(