Fix CosmosDB for latest API version

Signed-off-by: Bernd Verst <github@bernd.dev>

state/azure/cosmosdb/cosmosdb.go

@@ -81,7 +81,6 @@ type CosmosItem struct {
 const (
 	metadataPartitionKey = "partitionKey"
 	defaultTimeout       = 20 * time.Second
-	statusNotFound       = "NotFound"
 )
 
 // Policy that makes all queries cross-partition
@@ -228,8 +227,7 @@ func (c *StateStore) Get(ctx context.Context, req *state.GetRequest) (*state.Get
 	defer cancel()
 	readItem, err := c.client.ReadItem(readCtx, azcosmos.NewPartitionKeyString(partitionKey), req.Key, &options)
 	if err != nil {
-		var responseErr *azcore.ResponseError
-		if errors.As(err, &responseErr) && responseErr.ErrorCode == "NotFound" {
+		if isNotFoundError(err) {
 			return &state.GetResponse{}, nil
 		}
 		return nil, err
@@ -685,14 +683,16 @@ func populatePartitionMetadata(key string, requestMetadata map[string]string) st
 }
 
 func isNotFoundError(err error) bool {
+	fmt.Printf("Check isNotFoundError: %v\n", err)
 	if err == nil {
 		return false
 	}
 
 	if requestError, ok := err.(*azcore.ResponseError); ok {
-		if requestError.ErrorCode == statusNotFound {
+		if requestError.StatusCode == 404 {
 			return true
 		}
+		// we previously checked the error code, but unfortunately this is not stable between API versions
 	}
 
 	return false