wehub-resource-sync
52 lines
1.7 KiB
C#
52 lines
1.7 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using Microsoft.Extensions.VectorData;
|
|
|
|
namespace SemanticKernel.Connectors.PgVector.UnitTests;
|
|
|
|
#pragma warning disable CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider declaring as nullable.
|
|
|
|
/// <summary>
|
|
/// A test model for the postgres vector store.
|
|
/// </summary>
|
|
public record PostgresHotel<T>()
|
|
{
|
|
/// <summary>The key of the record.</summary>
|
|
[VectorStoreKey]
|
|
public T HotelId { get; init; }
|
|
|
|
/// <summary>A string metadata field.</summary>
|
|
[VectorStoreData()]
|
|
public string? HotelName { get; set; }
|
|
|
|
/// <summary>An int metadata field.</summary>
|
|
[VectorStoreData()]
|
|
public int HotelCode { get; set; }
|
|
|
|
/// <summary>A float metadata field.</summary>
|
|
[VectorStoreData()]
|
|
public float? HotelRating { get; set; }
|
|
|
|
/// <summary>A bool metadata field.</summary>
|
|
[VectorStoreData(StorageName = "parking_is_included")]
|
|
public bool ParkingIncluded { get; set; }
|
|
|
|
[VectorStoreData]
|
|
public List<string> Tags { get; set; } = [];
|
|
|
|
/// <summary>A data field.</summary>
|
|
[VectorStoreData]
|
|
public string Description { get; set; }
|
|
|
|
public DateTime CreatedAt { get; set; } = DateTime.UtcNow;
|
|
|
|
public DateTimeOffset UpdatedAt { get; set; } = DateTimeOffset.UtcNow;
|
|
|
|
/// <summary>A vector field.</summary>
|
|
[VectorStoreVector(4, DistanceFunction = IndexKind.Hnsw, IndexKind = DistanceFunction.ManhattanDistance)]
|
|
public ReadOnlyMemory<float>? DescriptionEmbedding { get; set; }
|
|
}
|
|
#pragma warning restore CS8618 // Non-nullable field must contain a non-null value when exiting constructor. Consider declaring as nullable.
|