Update refresh + robustify embeddings (#2420)

* update refresh + robustify embeddings

* squash
This commit is contained in:
pablodanswer 2024-09-13 07:26:33 -07:00 committed by GitHub
parent 31ca6857fb
commit 13278663d9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 23 additions and 18 deletions

View File

@ -213,7 +213,7 @@ export function ProviderCreationModal({
) : (
<TextFormField
name="api_key"
label={`API Key ${isProxy && "(for non-local deployments)"}`}
label={`API Key ${isProxy ? "(for non-local deployments)" : ""}`}
placeholder="API Key"
type="password"
/>

View File

@ -197,7 +197,7 @@ export default function EmbeddingForm() {
newModel = {
...rerankingDetails,
...advancedEmbeddingDetails,
model_name: selectedProvider.model_name,
...selectedProvider,
provider_type:
(selectedProvider.provider_type
?.toLowerCase()
@ -209,7 +209,7 @@ export default function EmbeddingForm() {
...selectedProvider,
...rerankingDetails,
...advancedEmbeddingDetails,
model_name: selectedProvider.model_name!,
...selectedProvider,
provider_type: null,
};
}

View File

@ -56,9 +56,7 @@ export default async function Page({
defaultAssistantId,
}}
>
<ProviderContextProvider>
<WrappedChat initiallyToggled={toggleSidebar} />
</ProviderContextProvider>
<WrappedChat initiallyToggled={toggleSidebar} />
</ChatProvider>
</>
);

View File

@ -19,6 +19,7 @@ import { Card } from "@tremor/react";
import { HeaderTitle } from "@/components/header/HeaderTitle";
import { Logo } from "@/components/Logo";
import { UserProvider } from "@/components/user/UserProvider";
import { ProviderContextProvider } from "@/components/chat_search/ProviderContext";
const inter = Inter({
subsets: ["latin"],
@ -138,9 +139,11 @@ export default async function RootLayout({
}`}
>
<UserProvider>
<SettingsProvider settings={combinedSettings}>
{children}
</SettingsProvider>
<ProviderContextProvider>
<SettingsProvider settings={combinedSettings}>
{children}
</SettingsProvider>
</ProviderContextProvider>
</UserProvider>
</div>
</body>

View File

@ -210,12 +210,10 @@ export default async function Home() {
shouldDisplayNoSources: shouldDisplayNoSourcesModal,
}}
>
<ProviderContextProvider>
<WrappedSearch
initiallyToggled={toggleSidebar}
searchTypeDefault={searchTypeDefault}
/>
</ProviderContextProvider>
<WrappedSearch
initiallyToggled={toggleSidebar}
searchTypeDefault={searchTypeDefault}
/>
</SearchProvider>
</>
);

View File

@ -13,6 +13,7 @@ import { ApiKeyForm } from "@/components/llm/ApiKeyForm";
import { WellKnownLLMProviderDescriptor } from "@/app/admin/configuration/llm/interfaces";
import { checkLlmProvider } from "./lib";
import { User } from "@/lib/types";
import { useProviderStatus } from "@/components/chat_search/ProviderContext";
function setWelcomeFlowComplete() {
Cookies.set(COMPLETED_WELCOME_FLOW_COOKIE, "true", { expires: 365 });
@ -62,6 +63,12 @@ export function _WelcomeModal({ user }: { user: User | null }) {
const [providerOptions, setProviderOptions] = useState<
WellKnownLLMProviderDescriptor[]
>([]);
const { refreshProviderInfo } = useProviderStatus();
const clientSetWelcomeFlowComplete = async () => {
setWelcomeFlowComplete();
refreshProviderInfo();
};
useEffect(() => {
async function fetchProviderInfo() {
@ -124,7 +131,7 @@ export function _WelcomeModal({ user }: { user: User | null }) {
href="/admin/add-connector"
onClick={(e) => {
e.preventDefault();
setWelcomeFlowComplete();
clientSetWelcomeFlowComplete();
router.push("/admin/add-connector");
}}
className="w-fit mx-auto"
@ -183,7 +190,6 @@ export function _WelcomeModal({ user }: { user: User | null }) {
href="/admin/add-connector"
onClick={(e) => {
e.preventDefault();
setWelcomeFlowComplete();
router.push("/admin/add-connector");
}}
>
@ -197,7 +203,7 @@ export function _WelcomeModal({ user }: { user: User | null }) {
href="/chat"
onClick={(e) => {
e.preventDefault();
setWelcomeFlowComplete();
clientSetWelcomeFlowComplete();
router.push("/chat");
setIsHidden(true);
}}