Compare commits

...

2 Commits

Author SHA1 Message Date
Richard Kuo (Onyx)
4c753c0d4e fix disabled appearance on continue 2025-03-18 20:34:46 -07:00
Richard Kuo (Onyx)
ed027f0406 improve logging on embeddings 2025-03-18 20:33:54 -07:00
3 changed files with 57 additions and 59 deletions

View File

@@ -412,6 +412,8 @@ async def embed_text(
reduced_dimension: int | None,
gpu_type: str = "UNKNOWN",
) -> list[Embedding]:
embeddings: list[Embedding]
if not all(texts):
logger.error("Empty strings provided for embedding")
raise ValueError("Empty strings are not allowed for embedding.")
@@ -463,11 +465,17 @@ async def embed_text(
logger.error(error_message)
raise ValueError(error_message)
num_floats = 0
for embedding in embeddings:
num_floats += len(embedding)
elapsed = time.monotonic() - start
logger.info(
f"event=embedding_provider "
f"texts={len(texts)} "
f"chars={total_chars} "
f"embeddings={len(embeddings)} "
f"embedding_floats={num_floats} "
f"provider={provider_type} "
f"elapsed={elapsed:.2f}"
)
@@ -493,6 +501,10 @@ async def embed_text(
for embedding in embeddings_vectors
]
num_floats = 0
for embedding in embeddings:
num_floats += len(embedding)
elapsed = time.monotonic() - start
logger.info(
f"Successfully embedded {len(texts)} texts with {total_chars} total characters "
@@ -502,6 +514,8 @@ async def embed_text(
f"event=embedding_model "
f"texts={len(texts)} "
f"chars={total_chars} "
f"embeddings={len(embeddings)} "
f"embedding_floats={num_floats} "
f"model={model_name} "
f"gpu={gpu_type} "
f"elapsed={elapsed:.2f}"

View File

@@ -449,7 +449,48 @@ export default function AddConnector({
{formStep == 0 && (
<CardSection>
<Title className="mb-2 text-lg">Select a credential</Title>
<div className="flex items-center justify-between mb-2">
<Title className="text-lg">Select a credential</Title>
<div className="flex gap-4">
<Button
variant="success-reverse"
className="text-sm"
onClick={async () => {
if (oauthDetails && oauthDetails.oauth_enabled) {
if (oauthDetails.additional_kwargs.length > 0) {
setCreateCredentialFormToggle(true);
} else {
const redirectUrl =
await getConnectorOauthRedirectUrl(connector, {});
if (redirectUrl) {
window.location.href = redirectUrl;
} else {
setCreateCredentialFormToggle((prev) => !prev);
}
}
} else {
setCreateCredentialFormToggle((prev) => !prev);
}
}}
>
Enter Manually
</Button>
{oauthSupportedSources.includes(connector) &&
(NEXT_PUBLIC_CLOUD_ENABLED || NEXT_PUBLIC_TEST_ENV) && (
<Button
variant="success-reverse"
onClick={handleAuthorize}
disabled={isAuthorizing}
hidden={!isAuthorizeVisible}
>
{isAuthorizing
? "Authorizing..."
: `Create via ${getSourceDisplayName(connector)}`}
</Button>
)}
</div>
</div>
{connector == ValidSources.Gmail ? (
<GmailMain />
@@ -464,63 +505,6 @@ export default function AddConnector({
onDeleteCredential={onDeleteCredential}
onSwitch={onSwap}
/>
{!createCredentialFormToggle && (
<div className="mt-6 flex space-x-4">
{/* Button to pop up a form to manually enter credentials */}
<Button
variant="secondary"
className="mt-6 text-sm mr-4"
onClick={async () => {
if (oauthDetails && oauthDetails.oauth_enabled) {
if (oauthDetails.additional_kwargs.length > 0) {
setCreateCredentialFormToggle(true);
} else {
const redirectUrl =
await getConnectorOauthRedirectUrl(
connector,
{}
);
// if redirect is supported, just use it
if (redirectUrl) {
window.location.href = redirectUrl;
} else {
setCreateCredentialFormToggle(
(createConnectorToggle) =>
!createConnectorToggle
);
}
}
} else {
setCreateCredentialFormToggle(
(createConnectorToggle) =>
!createConnectorToggle
);
}
}}
>
Create New
</Button>
{/* Button to sign in via OAuth */}
{oauthSupportedSources.includes(connector) &&
(NEXT_PUBLIC_CLOUD_ENABLED ||
NEXT_PUBLIC_TEST_ENV) && (
<Button
variant="navigate"
onClick={handleAuthorize}
className="mt-6 "
disabled={isAuthorizing}
hidden={!isAuthorizeVisible}
>
{isAuthorizing
? "Authorizing..."
: `Authorize with ${getSourceDisplayName(
connector
)}`}
</Button>
)}
</div>
)}
{createCredentialFormToggle && (
<Modal
className="max-w-3xl rounded-lg"

View File

@@ -65,7 +65,7 @@ const NavigationRow = ({
<div className="flex justify-end">
{formStep === 0 && (
<SquareNavigationButton
className="bg-blue-400 text-white py-2.5 px-3.5 disabled:bg-blue-200"
className="bg-blue-400 text-white py-2.5 px-3.5 disabled:bg-gray-600"
disabled={!activatedCredential}
onClick={nextFormStep}
>